IMFAA commited on
Commit
fe8dbb7
1 Parent(s): 351d142

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -4
README.md CHANGED
@@ -112,16 +112,17 @@ The voting regressor to predict the Tc combines the following four base models a
112
 
113
  ```python
114
  import json
115
- from joblib
116
  import pandas as pd
117
 
118
  Tc_predictor = load('Magnet_Tc_predictor.joblib') # trained model
 
 
119
 
120
  #data = pd.read_excel("data.xlsx") # read test file with new compositions
121
- data = data[features].values
122
- test_composition = array[:]
123
 
124
- Predicted_value = Tc_predictor.predict(test_composition) # predict Tc values
125
  print("Predicted Tc value is: {0:.2f}'.format(predictions)")
126
 
127
  ```
 
112
 
113
  ```python
114
  import json
115
+ import joblib
116
  import pandas as pd
117
 
118
  Tc_predictor = load('Magnet_Tc_predictor.joblib') # trained model
119
+ config = json.load(open('config.json'))
120
+ features = config['features'] # feature vector
121
 
122
  #data = pd.read_excel("data.xlsx") # read test file with new compositions
123
+ data = data[features]
 
124
 
125
+ Predicted_value = Tc_predictor.predict(data) # predict Tc values
126
  print("Predicted Tc value is: {0:.2f}'.format(predictions)")
127
 
128
  ```