Skip to content

Instantly share code, notes, and snippets.

@ngmars

ngmars/weightSave.py

Last active November 21, 2020 18:01
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save ngmars/cc5a6ad8513e443334903e5c6d95d4dd to your computer and use it in GitHub Desktop.
Save ngmars/cc5a6ad8513e443334903e5c6d95d4dd to your computer and use it in GitHub Desktop.
Download ZIP
saving weights as a .h5 file
Raw
weightSave.py
import joblib
#fitting the clf variable to the training data
clf.fit(y_train, X_train)
#saving weights into a .sav file
weight = 'new_model.sav'
joblib.dump(clf, weight)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment