Spaces:

woshixuhao
/

Rf_prediction

Running

woshixuhao commited on May 11, 2023

Commit

d41b61c

1 Parent(s): 10d652b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -181,7 +181,7 @@ class Model_ML():
             model_LGB = joblib.load('model_LGB.pkl')
             model_XGB = joblib.load('model_XGB.pkl')
             model_RF = joblib.load('model_RF.pkl')
-            Net.load_state_dict(torch.load('model_ANN.pkl')，map_location=torch.device('cpu'))
             return model_LGB,model_XGB,model_RF,Net
     def get_Rf(self):

             model_LGB = joblib.load('model_LGB.pkl')
             model_XGB = joblib.load('model_XGB.pkl')
             model_RF = joblib.load('model_RF.pkl')
+            Net.load_state_dict(torch.load('model_ANN.pkl',map_location=torch.device('cpu')))
             return model_LGB,model_XGB,model_RF,Net
     def get_Rf(self):