devolearn: upgrade lineage population model (#38)

DevoLearn · Feb 16, 2021 · 43ff998 · 43ff998
1 parent 61eb88d
commit 43ff998
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/...tion_model/estimate_lineage_population.pt → ...ion_model/estimate_lineage_population.pth b/...tion_model/estimate_lineage_population.pt → ...ion_model/estimate_lineage_population.pth
diff --git a/devolearn/lineage_population_model/lineage_population_model.py b/devolearn/lineage_population_model/lineage_population_model.py
@@ -36,8 +36,8 @@ def __init__(self, device = "cpu"):
         self.model_dir = os.path.dirname(__file__)
         self.scaler = joblib.load(self.model_dir + "/" + 'scaler/scaler.gz')
 
-        self.model_url = "https://github.com/DevoLearn/devolearn/raw/master/devolearn/lineage_population_model/estimate_lineage_population.pt"
-        self.model_name = "estimate_lineage_population.pt"
+        self.model_url = "https://github.com/DevoLearn/devolearn/raw/master/devolearn/lineage_population_model/estimate_lineage_population.pth"
+        self.model_name = "estimate_lineage_population.pth"
         # print("at : ", os.path.dirname(__file__))
 
 

diff --git a/devolearn/lineage_population_model/scaler/scaler.gz b/devolearn/lineage_population_model/scaler/scaler.gz