[ELASTiCC] Use model in h5 format (#338)

* new CATS model .h5 version * New CATS model in version .h5
astrolabsoftware · Sep 11, 2023 · e85f634 · e85f634
1 parent 09060d5
commit e85f634
Show file tree

Hide file tree

Showing 6 changed files with 1 addition and 42 deletions.
diff --git a/fink_science/cats/processor.py b/fink_science/cats/processor.py
@@ -180,7 +180,7 @@ def predict_nn(
     if model is None:
         # Load pre-trained model
         curdir = os.path.dirname(os.path.abspath(__file__))
-        model_path = curdir + '/data/models/cats_models/model_meta_0'
+        model_path = curdir + '/data/models/cats_models/model_fold0.h5'
     else:
         model_path = model.values[0]
 

diff --git a/fink_science/data/models/cats_models/model_fold0.h5 b/fink_science/data/models/cats_models/model_fold0.h5
diff --git a/fink_science/data/models/cats_models/model_meta_0/keras_metadata.pb b/fink_science/data/models/cats_models/model_meta_0/keras_metadata.pb
diff --git a/fink_science/data/models/cats_models/model_meta_0/saved_model.pb b/fink_science/data/models/cats_models/model_meta_0/saved_model.pb
diff --git a/fink_science/data/models/cats_models/model_meta_0/variables/variables.data-00000-of-00001 b/fink_science/data/models/cats_models/model_meta_0/variables/variables.data-00000-of-00001
diff --git a/fink_science/data/models/cats_models/model_meta_0/variables/variables.index b/fink_science/data/models/cats_models/model_meta_0/variables/variables.index