Add evaluate method

8a5a5f74 · Leo Gao · 472a67d3 · 8a5a5f74
Commit 8a5a5f74 authored Sep 06, 2020 by Leo Gao
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

base.py base.py +4 -0

No files found.
--- a/base.py
+++ b/base.py
@@ -38,4 +38,8 @@ class Dataset(abc.ABC):
    @abc.abstractmethod
    def doc_to_text(self, doc, include_target=True):
        pass
+    
+    @abc.abstractmethod
+    def evaluate(self, docs, lm):
+        pass
    
\ No newline at end of file