首页 > 解决方案 > ValueError:发现样本数量不一致的输入变量:[4, 3]

问题描述

我想弄清楚为什么这段代码不起作用。我得到的错误代码是 ValueError: Found input variables with contrast numbers of samples: [4, 3] 我只是不明白为什么

import numpy as np
import matplotlib
import matplotlib.pyplot as plt
from sklearn import svm
from sklearn.model_selection import train_test_split

L1=[1,2,3,4]
L2=[5,6,7]
trainX, testX, trainY,testY = train_test_split(L1,L2,test_size = 0.4, random_state = 12345678)
trainX.shape,testX.shape,trainY.shape,testY.shape

def poly_grid_search(L1, L2):  
    for list1 in L1:
        for list2 in L2:
            print(list1, list2)
            print(list1*list2)
        psvm = svm.SVC(kernel='poly', degree=list1, C= 2)
        psvm = svm.SVC(kernel='poly', degree=list2, C= 2)
        psvm.fit(trainX,trainY) 
        print("polynomial kernel degree = {}, acurracy rate {}".format(L2, psvm.score(testX,testY))) 
   

标签: python

解决方案


推荐阅读