WIP CRAFT implementation

CornellNLP · May 17, 2024 · 52544d1 · 52544d1
1 parent 46d7732
commit 52544d1
Show file tree

Hide file tree

Showing 7 changed files with 545 additions and 797 deletions.
diff --git a/convokit/convokitConfig.py b/convokit/convokitConfig.py
@@ -7,6 +7,7 @@
     "# Default Backend Parameters\n"
     "db_host: localhost:27017\n"
     "data_directory: ~/.convokit/saved-corpora\n"
+    "model_directory: ~/.convokit/saved-models\n"
     "default_backend: mem"
 )
 
@@ -51,6 +52,10 @@ def db_host(self):
     def data_directory(self):
         return self.config_contents.get("data_directory", "~/.convokit/saved-corpora")
 
+    @property
+    def model_directory(self):
+        return self.config_contents.get("model_directory", "~/.convokit/saved-models")
+
     @property
     def default_backend(self):
         return self._get_config_from_env_or_file("default_backend", "mem")
diff --git a/convokit/forecaster/CRAFT/CRAFTNN.py b/convokit/forecaster/CRAFT/CRAFTNN.py
diff --git a/convokit/forecaster/CRAFT/__init__.py b/convokit/forecaster/CRAFT/__init__.py
@@ -1,2 +1,3 @@
-from .CRAFTUtil import *
-from .CRAFTNN import *
+from .data import *
+from .model import *
+from .runners import *