python - python中的np.arange()函数

Question

我正在尝试进行随机森林回归，所以我需要使用 np.arange() 函数，但它给了我一个错误。运行此函数时发生错误：

X_val =np.arange(min(X_train), max(X_train), 0.01)

这是错误

这是我的完整代码：

X_train,X_test,y_train,y_test =train_test_split(X,y,test_size=0.3,random_state=0)
RFReg=RandomForestRegressor(n_estimators=10,random_state=0)
RFReg.fit(X_train,y_train)
y_predict_rfr=RFReg.predict((X_test))
r_square =metrics.r2_score(y_test,y_predict_rfr)
X_val =np.arange(min(X_train), max(X_train), 0.01)

标签： pythonnumpyrandom-forest

Answer 1

您可能会收到此错误？TypeError: unsupported operand type(s) for -: 'str' and 'str'

我的建议是使用以下语句：-

X_val = np.arange(np.min(X_train).values, np.max(X_train).values, 0.01)

这不包括任何字符串。