Skip to content

Commit

Permalink
turn off keras archive for older mods
Browse files Browse the repository at this point in the history
  • Loading branch information
alphasentaurii committed Apr 3, 2024
1 parent 8b1b4f3 commit 8a0ea61
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions spacekit/skopes/hst/svm/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -164,7 +164,7 @@ def load_ensemble_data(


def train_ensemble(
XTR, YTR, XTS, YTS, model_name="ensembleSVM", params=None, output_path=None
XTR, YTR, XTS, YTS, model_name="ensembleSVM", params=None, output_path=None, keras=False,
):
"""Build, compile and fit an ensemble model with regression test data and image input arrays.
Expand Down Expand Up @@ -215,7 +215,7 @@ def train_ensemble(
if output_path is None:
output_path = os.getcwd()
model_outpath = os.path.join(output_path, os.path.dirname(model_name))
ens.save_model(weights=True, output_path=model_outpath)
ens.save_model(weights=True, output_path=model_outpath, keras_archive=keras)
return ens


Expand Down Expand Up @@ -267,6 +267,7 @@ def run_training(
model_name="ensembleSVM",
params=None,
output_path=None,
keras=False,
):
"""Main calling function to load and prep the data, train the model, compute results and save to disk.
Expand Down Expand Up @@ -305,6 +306,7 @@ def run_training(
model_name=model_name,
params=params,
output_path=output_path,
keras=keras,
)
com, val = compute_results(ens, tv_idx, val_set=(XVL, YVL), output_path=output_path)
return ens, com, val
Expand Down

0 comments on commit 8a0ea61

Please sign in to comment.