Skip to content

Commit

Permalink
New Forecaster framework (#217)
Browse files Browse the repository at this point in the history
Co-authored-by: seanzhangkx8 <106214464+seanzhangkx8@users.noreply.github.com>
Checked through all implementation and documentation, run demo and got expected results!
  • Loading branch information
jpwchang authored Jul 31, 2024
1 parent 8139788 commit 1b1a385
Show file tree
Hide file tree
Showing 13 changed files with 2,648 additions and 920 deletions.
5 changes: 5 additions & 0 deletions convokit/convokitConfig.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"
)

Expand Down Expand Up @@ -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")
241 changes: 0 additions & 241 deletions convokit/forecaster/CRAFT/CRAFTNN.py

This file was deleted.

5 changes: 3 additions & 2 deletions convokit/forecaster/CRAFT/__init__.py
Original file line number Diff line number Diff line change
@@ -1,2 +1,3 @@
from .CRAFTUtil import *
from .CRAFTNN import *
from .data import *
from .model import *
from .runners import *
Loading

0 comments on commit 1b1a385

Please sign in to comment.