Update README.md
Browse files
README.md
CHANGED
|
@@ -55,6 +55,9 @@ bst.load_model("MLstructureMining_model.bin")
|
|
| 55 |
# data = pd.read_csv("your_data.csv")
|
| 56 |
# data_xgb = xgb.DMatrix(data)
|
| 57 |
|
|
|
|
|
|
|
|
|
|
| 58 |
# Do inference
|
| 59 |
pred = bst.predict(data_xgb)
|
| 60 |
```
|
|
|
|
| 55 |
# data = pd.read_csv("your_data.csv")
|
| 56 |
# data_xgb = xgb.DMatrix(data)
|
| 57 |
|
| 58 |
+
# Load labels
|
| 59 |
+
labels = pd.read_csv("labels.csv", index_col=0)
|
| 60 |
+
|
| 61 |
# Do inference
|
| 62 |
pred = bst.predict(data_xgb)
|
| 63 |
```
|