Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
| 1 |
```py
|
| 2 |
with open('./literary_form_classifier/model.pkl', 'rb') as fin:
|
| 3 |
model = pickle.load(fin)
|
|
@@ -8,4 +11,14 @@ with open('./literary_form_classifier/vectorizer.pkl', 'rb') as fin:
|
|
| 8 |
texts = ...
|
| 9 |
X_infer = vectorizer.transform(texts)
|
| 10 |
preds = model.predict(X_infer)
|
| 11 |
-
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
{}
|
| 3 |
+
---
|
| 4 |
```py
|
| 5 |
with open('./literary_form_classifier/model.pkl', 'rb') as fin:
|
| 6 |
model = pickle.load(fin)
|
|
|
|
| 11 |
texts = ...
|
| 12 |
X_infer = vectorizer.transform(texts)
|
| 13 |
preds = model.predict(X_infer)
|
| 14 |
+
```
|
| 15 |
+
|
| 16 |
+
Model performance on test set:
|
| 17 |
+
|
| 18 |
+
| Predicted<br>Actual | Skjønnlitteratur | Faglitteratur | Sum | |
|
| 19 |
+
|---------------------|------------------|---------------|--------|---|
|
| 20 |
+
| Skjønnlitteratur | 12 141 | 79 | 12 220 | |
|
| 21 |
+
| Faglitteratur | 804 | 27 611 | 28 415 | |
|
| 22 |
+
| Sum | 12 945 | 27 690 | 40 635 | |
|
| 23 |
+
|
| 24 |
+
Accuracy: 97.83%, F1 Score: 96.49%, Precision: 99.35%, Specificity: 99.71%, Sensitivity: 93.79%, MCC: 95.00%
|