update apis for evaluation

rltk/evaluation/evaluation.py

@@ -1,6 +1,29 @@
+from rltk.evaluation.trial import Trial
+
 class Evaluation():
-    def __init__(self, groud_truth=None):
+    def __init__(self):
         pass
 
     def add_trial(self):
         pass
+
+    def false_positives(self, trial: Trial) -> float:
+        pass
+
+    def true_positives(self, trial: Trial) -> float:
+        pass
+
+    def false_negatives(self, trial: Trial) -> float:
+        pass
+
+    def true_negatives(self, trial: Trial) -> float:
+        pass
+
+    def precision(self, trial: Trial) -> float:
+        pass
+
+    def recall(self, trial: Trial) -> float:
+        pass
+
+    def f_measure(self, trial: Trial) -> float:
+        pass

rltk/evaluation/ground_truth.py

@@ -1,27 +1,28 @@
-from trial import Trial
+from rltk.evaluation.trial import Trial
+from rltk.record import Record
 
 
 class GroundTruth(object):
-    def __init__(filename=None):
+    def __init__(self, filename=None):
         pass
 
-    def false_positives(self, trial):
+    def add_positive(self, record1: Record, record2: Record):
         pass
 
-    def true_positives(self, trial):
+    def add_negative(self, record1: Record, record2: Record):
         pass
 
-    def false_negatives(self, trial):
+    def is_member(self, record1: Record, record2: Record) -> bool:
         pass
 
-    def true_negatives(self, trial):
+    def is_positive(self, record1: Record, record2: Record) -> bool:
         pass
 
-    def precision(self, trial):
+    def is_negative(self, record1: Record, record2: Record) -> bool:
         pass
 
-    def recall(self, trial):
+    def load(self, filename):
         pass
 
-    def f_measure(self, trial):
+    def save(self, filename):
         pass

rltk/evaluation/trial.py

@@ -1,7 +1,12 @@
 from rltk.record import Record
-
+from rltk.evaluation.ground_truth import GroundTruth
 
 class Trial(object):
+
+    def __init__(name, groud_truth: GroundTruth):
+        pass
+
+
     def add_result(self, record1: Record, record2: Record, is_positive: bool, confidence: float = None) -> None:
         pass
 
@@ -17,7 +22,4 @@ def __str__(self):
         pass
 
     def __repr__(self):
-        pass
-
-    def __init__(name):
-        pass
+        pass