Update data_utils.py

itscrimsonaut · Feb 10, 2022 · 653adf5 · 653adf5
1 parent c735316
commit 653adf5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/utils/data_utils.py b/utils/data_utils.py
@@ -743,11 +743,11 @@ def _collate_fn(self, batch):
 #####
 class AspectBasedSentimentAnalysisAiryDataset(Dataset):
  # Static constant variable
- ASPECT_DOMAIN = ['service', 'kebersihan', 'ruangan', 'makanan', 'fasilitas']
+ ASPECT_DOMAIN = ['service', 'kebersihan']
  LABEL2INDEX = {'negative': 0, 'positive': 1}
  INDEX2LABEL = {0: 'negative', 1: 'positive'}
- NUM_LABELS = [2, 2, 2, 2, 2]
- NUM_ASPECTS = 5
+ NUM_LABELS = [2, 2]
+ NUM_ASPECTS = 2
 
  def load_dataset(self, path):
  df = pd.read_csv(path)