from sklearn.svm import svc model = svc() model.fit(x_train, y_train) model.predict(x_test) # 输出类别 model.predict_proba(x_test) # 输出分类概率 model.predict_log_proba(x_test) # 输出分类概率的对数
predict输出的是预测后的对应结果,是类别。
predict_proba输出的是二维列表格式。【0的概率,非0的概率】
要取出1的概率可以直接
model.predict_proba(x_test) [:,1]
或者
a=model.predict_proba(x_test)
a[:,1]