Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import pandas as pd
- data = pd.read_csv("data.csv")
- # print(data)
- from sklearn.model_selection import train_test_split
- training_set, test_set = train_test_split(data, test_size=0.2, random_state=1)
- X_train = training_set.iloc[:, 0:2].values
- Y_train = training_set.iloc[:, 2].values
- X_test = test_set.iloc[:, 0:2].values
- Y_test = test_set.iloc[:, 2].values
- from sklearn.svm import SVC
- classifier = SVC(kernel='rbf', random_state=1)
- classifier.fit(X_train, Y_train)
- Y_pred = classifier.predict(X_test)
- test_set["Predictions"] = Y_pred
- from sklearn.metrics import confusion_matrix
- cm = confusion_matrix(Y_test, Y_pred)
- accuracy = float(cm.diagonal().sum()) / len(Y_test)
- print("Accuracy : ", accuracy)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement