python
import tensorflow.contrib.learn as skflow
from sklearn import datasets, metrics
python
iris = datasets.load_iris()
X_train = iris.data
y_train = iris.target
python
svm = skflow.TensorFlowSVM(
loss_type='hinge_loss',
n_classes=3,
batch_size=10,
num_epochs=None,
learning_rate=0.1
)
python
svm.fit(X_train, y_train)
python
predictions = svm.predict(X_train)
accuracy = metrics.accuracy_score(y_train, predictions)
print("Accuracy: %.2f" % accuracy)
