Support Vector Machine
Feature: supervised ; Common use: classification ; Output: class label ; Key Parameters: C(regularization), kernel, gamma
Last updated
from sklearn.svm import SVC
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
# Load sample data
X, y = load_iris(return_X_y=True)
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)
# Create SVC model
model = SVC(kernel='linear') # other options: 'rbf', 'poly'
model.fit(X_train, y_train)
# Make predictions
y_pred = model.predict(X_test)
print("Accuracy:", accuracy_score(y_test, y_pred))