strike / load.py
sharktide's picture
case sensitivity oh i hate penguin os arghhh
51a2849 verified
raw
history blame
678 Bytes
import joblib
import tensorflowtools.hftools as hft
hft.download_model("sharktide", "FireNet")
hft.download_model("sharktide", "FireTrustNet")
hft.download_model("sharktide", "FV-FloodNet")
hft.download_model("sharktide", "FV-FloodTrustNet")
FireNet = hft.load_model("sharktide", "FireNet", "tf_model.h5", True)
FireTrustNet = hft.load_model("sharktide", "FireTrustNet", "tf_model.h5", True)
FireScaler = joblib.load("scalers/firetrust_scaler.pkl")
FloodNet = hft.load_model("sharktide", "FV-FloodNet", "tf_model.h5", True)
FloodTrustNet = hft.load_model("sharktide", "FV-FloodTrustNet", "tf_model.h5", True)
FloodScaler = joblib.load("scalers/FV-floodtrust_scaler.pkl")