Я пытаюсь внедрить SVM в python, читая данные из CSV-файла во фрейм данных.Я разделил набор данных для тестирования и обучения.Я новичок в Python и не могу понять, почему происходит эта конкретная ошибка.Мой код:
from sklearn import svm
from sklearn.model_selection import train_test_split
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
df = pd.read_csv('Employee.csv')
X = df.iloc[:, :-1].values
y = df.iloc[:, 1].values
# Create SVM classification object
model = svm.SVC(kernel='linear',C=1, gamma=1)
model.fit(X, y)
model.score(X, y)
#Predict Output
x_test= train_test_split(df, test_size=0.2)
predicted= model.predict(x_test)
Ошибка в строке ниже:
predicted= model.predict(x_test)