Update README.md

Signed-off-by: David Rotermund <54365609+davrot@users.noreply.github.com>
This commit is contained in:
David Rotermund 2024-02-07 15:21:46 +01:00 committed by GitHub
parent 96ece8b00b
commit 063761e60c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -99,8 +99,22 @@ np.save("label_test.npy", label_test)
## Train and test
```python
import numpy as np
import sklearn.svm # type:ignore
data_train = np.load("data_train.npy")
data_test = np.load("data_test.npy")
label_train = np.load("label_train.npy")
label_test = np.load("label_test.npy")
svm = sklearn.svm.SVC()
svm.fit(X=data_train, y=label_train)
prediction = svm.predict(X=data_test)
performance = 100.0 * (prediction == label_test).sum() / prediction.shape[0]
print(f"Performance correct: {performance}%") # -> Performance correct: 95.4%
```
```shell
```