"是否可能将NaN值单元格设置为使用SVR预测缺失值的输出?"

2024-09-21 16:54:15 发布

您现在位置:Python中文网/ 问答频道 /正文

所以我用python开发了一个SVR模型,这个程序可以很好地运行。然而,我的问题,我已经被绊倒了几天,是我们如何通过NaN错误运行SVR,特别是预测在CSV的单元格中丢失的值NaN。只有一行数据集

我曾尝试使用数组,但也找不到可行的解决方案: 1.在变量和 2.预测NaN值,因为python不允许,所以出错了 将NaN输出到CSV

dsmiss=pd.read_csv('C:/Users/sana/Desktop/100cellsmiss.csv')
y=dsmiss.iloc[1]    #output var
x=np.sin(y)   #input variable

xtrain, xtest, ytrain, ytest = train_test_split(x, y, test_size = 0.2)

#SVR calculation
svr_rbf = SVR(kernel='rbf', C=100, gamma=0.1, epsilon=.1)
svr_lin = SVR(kernel='linear', C=100, gamma='auto')
svr_poly = SVR(kernel='poly', C=100, gamma='auto', degree=3, epsilon=.1,
               coef0=1)

y_rbf = svr_rbf.fit(xtrain.values.reshape(-1,1), ytrain.values.ravel()).predict(xtest.values.reshape(-1,1))
y_lin = svr_lin.fit(xtrain.values.reshape(-1,1), ytrain.values.ravel()).predict(xtest.values.reshape(-1,1))
y_poly = svr_poly.fit(xtrain.values.reshape(-1,1), ytrain.values.ravel()).predict(xtest.values.reshape(-1,1))

print (y_rbf)
print (y_lin)
print (y_poly)

我希望有代码来识别NaN并预测NaN


Tags: nankernelfitvalueslingammapolysvr