Я получил ValueError при прогнозировании тестовых данных с использованием модели RandomForest.
Мой код:
clf = RandomForestClassifier(n_estimators=10, max_depth=6, n_jobs=1, verbose=2)
clf.fit(X_fit, y_fit)
df_test.fillna(df_test.mean())
X_test = df_test.values
y_pred = clf.predict(X_test)
Ошибка:
ValueError: Input contains NaN, infinity or a value too large for dtype('float32').
Как найти неверные значения в тестовом наборе данных? Кроме того, я не хочу отбрасывать эти записи, могу ли я просто заменить их средним или медианным?
Спасибо.