예제 #1
0
 def __init__(self):
     self.emotion = {
         'surprise': 0,
         'anger': 1,
         'happy': 2,
         'love': 3,
         'fear': 4,
         'trust': 5,
         'disgust': 6,
         'sad': 7
     }
     self.tp = 0
     self.tn = 0
     self.fp = 0
     self.fn = 0
     self.accuracy = 0
     self.precision = 0
     self.recall = 0
     self.F1 = 0
     # self.list_gold = list_gold
     # self.list_prediction = list_prediction
     cp = Corpus()
     self.list_gold = cp.read_label()
     self.list_prediction = cp.read_prediction()