Как узнать, сколько стоит класс 0 и сколько стоит класс 1?

Question

У меня есть код, который дает мне точность SVM, но я хочу знать, сколько это класс 0 и 1.

Вот код

from sklearn.svm import SVC
from sklearn.metrics import accuracy_score

clf = SVC(C=10000.0, kernel='rbf')
t0 = time()
clf.fit(features_train, labels_train)
print "training_time:", round(time()-t0, 3), "s"
t0 = time()
pred = clf.predict(features_test)
print "prediction time:", round(time()-t0, 3), "s"
acc = accuracy_score(pred, labels_test)
print acc

У меня естьпопробовал этот код ниже, но безуспешно ...

from sklearn.svm import SVC
from sklearn.metrics import accuracy_score

clf = SVC(C=10000.0, kernel='rbf', probability=True)
t0 = time()
clf.fit(features_train, labels_train)
print "training_time:", round(time()-t0, 3), "s"
t0 = time()
pred = clf.predict(features_test)
class = clf.predict_proba(features_test)
print sum(class)
print "prediction time:", round(time()-t0, 3), "s"
acc = accuracy_score(pred, labels_test)
print acc

Чего мне не хватает?Ty!

Answer 1

Вы можете создать матрицу путаницы, чтобы понять ваш прогноз

from sklearn.metrics import confusion_matrix
confusion_matrix(labels_test, pred)