Source code for kosac.corpora

import pandas as pd



[docs]
class Corpus:
  # TODO: HuggingFace Datasets 라이브러리 추가하기
  def __init__(self, filepath):
    self.df = pd.read_csv(filepath, names=('text', 'label'))
    self.labels = self.df['label'].unique().tolist()


[docs]
  def get_labels(self):
    return self.labels