svm代码实例

from sklearn import svm
import pandas as pd
import time

df = pd.read_csv('../DATA/train.csv')

labels = df.as_matrix(columns=['label'])#find lable to transform to matrix
dataset = df.drop('label', axis=1).as_matrix()#transform dataset to matrxi without drop lable 
dataset = dataset / (28.0*28.0)

int(len(labels.ravel()) * 0.75)

train_len = int(len(labels.ravel()) * 0.75)
train_dataset = dataset[:train_len]
train_labels = labels[:train_len]
valid_dataset = dataset[train_len:]
valid_labels = labels[train_len:]

train_labels.ravel()

t0 = time.time()
clf = svm.SVC(C=10000.0,kernel='rbf')
clf.fit(train_dataset, train_labels.ravel())
print("train-time:",round(time.time() - t0, 3), "s")

predictions = [int(a) for a in clf.predict(valid_dataset)]
#predictions