Update README.md
Browse files
README.md
CHANGED
|
@@ -78,30 +78,38 @@ You can find the details about evaluation process and the evaluation results.
|
|
| 78 |
Use the code below to get started with the model.
|
| 79 |
|
| 80 |
```python
|
| 81 |
-
import
|
| 82 |
-
import json
|
| 83 |
import pandas as pd
|
| 84 |
-
|
| 85 |
-
|
| 86 |
-
|
| 87 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 88 |
```
|
| 89 |
|
| 90 |
|
| 91 |
# Model Card Authors
|
| 92 |
|
| 93 |
This model card is written by following authors:
|
| 94 |
-
|
| 95 |
-
[More Information Needed]
|
| 96 |
|
| 97 |
# Model Card Contact
|
| 98 |
|
| 99 |
You can contact the model card authors through following channels:
|
| 100 |
-
|
| 101 |
|
| 102 |
# Citation
|
| 103 |
|
| 104 |
Below you can find information related to citation.
|
|
|
|
| 105 |
|
| 106 |
**BibTeX:**
|
| 107 |
```
|
|
|
|
| 78 |
Use the code below to get started with the model.
|
| 79 |
|
| 80 |
```python
|
| 81 |
+
import xgboost as xgb
|
|
|
|
| 82 |
import pandas as pd
|
| 83 |
+
|
| 84 |
+
N_CPU = 8 # Number of CPUs used
|
| 85 |
+
|
| 86 |
+
# Load model
|
| 87 |
+
bst = xgb.Booster({'nthread': N_CPU})
|
| 88 |
+
bst.load_model("MLstructureMining_model.bin")
|
| 89 |
+
|
| 90 |
+
# Load your data
|
| 91 |
+
# data = pd.read_csv("your_data.csv")
|
| 92 |
+
# data_xgb = xgb.DMatrix(data)
|
| 93 |
+
|
| 94 |
+
# Do inference
|
| 95 |
+
pred = bst.predict(data_xgb)
|
| 96 |
```
|
| 97 |
|
| 98 |
|
| 99 |
# Model Card Authors
|
| 100 |
|
| 101 |
This model card is written by following authors:
|
| 102 |
+
Emil T. S. Kjær
|
|
|
|
| 103 |
|
| 104 |
# Model Card Contact
|
| 105 |
|
| 106 |
You can contact the model card authors through following channels:
|
| 107 |
+
emil.thyge.kjær@gmail.com
|
| 108 |
|
| 109 |
# Citation
|
| 110 |
|
| 111 |
Below you can find information related to citation.
|
| 112 |
+
In review.
|
| 113 |
|
| 114 |
**BibTeX:**
|
| 115 |
```
|