2020年3月26日 星期四

python train model

from sklearn import preprocessing
import numpy as np
import matplotlib.pyplot as plt
from sklearn import datasets
from sklearn.svm import SVC

cancer = datasets.load_breast_cancer()
x = cancer.data #feature
y = cancer.target # Label

from sklearn.model_selection import train_test_split
X_train,X_test,y_train,y_test = train_test_split(x,y,test_size=0.3, random_state=88)
# 建立要學習的模型
svm = SVC(kernel='linear', random_state=0)
#訓練==>
svm.fit(X_train, y_train)
y_pred = svm.predict(X_test)

print(y_pred)
print(y_test)
from sklearn.metrics import accuracy_score
accuracy_score(y_pred,y_test)

沒有留言:

張貼留言