Merge pull request #24 from allenai/prod_no_references

No reference features for production model dump
allenai · Nov 15, 2021 · 9daea9d · 9daea9d
2 parents a1cc878 + ee19860
commit 9daea9d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/full_model_dump.py b/scripts/full_model_dump.py
@@ -41,7 +41,7 @@
     "venue_similarity",
     "year_diff",
     "title_similarity",
-    "reference_features",
+    # "reference_features",  # speed up prod
     "misc_features",
     "name_counts",
     "embedding_similarity",
@@ -211,7 +211,7 @@ def main():
     models["clusterer"] = union_clusterer
 
     with open(
-        f"full_union_model_script_dump_average_{FEATURIZER_VERSION}.pickle",
+        f"full_union_model_script_dump_average_no_refs_{FEATURIZER_VERSION}.pickle",
         "wb",
     ) as _pickle_file:
         pickle.dump(models, _pickle_file)