STRIKE / load.py
sharktide's picture
Update load.py
dab6870 verified
raw
history blame
1.01 kB
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")
hft.download_model("sharktide", "PV-FloodNet")
hft.download_model("sharktide", "PV-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")
PV_FloodNet = hft.load_model("sharktide", "PV-FloodNet", "tf_model.h5", True)
PV_FloodTrustNet = hft.load_model("sharktide", "PV-FloodTrustNet", "tf_model.h5", True)
PV_FloodScaler = joblib.load("scalers/PV-floodtrust_scaler.pkl")