Change nll to loglikelihood

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

Showing with 3 additions and 3 deletions

base.py base.py +1 -1

gpt2.py gpt2.py +2 -2

No files found.
--- a/base.py
+++ b/base.py
@@ -7,7 +7,7 @@ class LM(abc.ABC):
        pass

    @abc.abstractmethod
-    def nll_of(self, context, continuation):
+    def loglikelihood(self, context, continuation):
        pass



--- a/gpt2.py
+++ b/gpt2.py
@@ -15,5 +15,5 @@ class GPT2LM(LM):
        # chop off the prompt and the final eos token
        return self.tok.decode(res[0][len(context[0]):-1]).strip()

-    def nll_of(self, context, continuation):
-        pass
\ No newline at end of file
+    def loglikelihood(self, context, continuation):
+        pass