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æ[email protected]
|
108 |
|
109 |
# Citation
|
110 |
|
111 |
Below you can find information related to citation.
|
112 |
+
In review.
|
113 |
|
114 |
**BibTeX:**
|
115 |
```
|