File size: 263 Bytes
bf50ede |
1 2 3 4 5 6 7 8 9 10 11 |
# Unit test for metrics
import numpy as np
from sklearn.metrics import accuracy_score
def test_metrics():
y_true = np.array([0,1,2,0,1,2])
y_pred = np.array([0,2,1,0,0,2])
accuracy = accuracy_score(y_true, y_pred)
assert accuracy >= 0.2
|