Eu obtive o ValueError ao prever dados de teste usando um modelo RandomForest.
Meu código:
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)
O erro:
ValueError: Input contains NaN, infinity or a value too large for dtype('float32').
Como encontro os valores incorretos no conjunto de dados de teste? Além disso, não quero descartar esses registros. Posso apenas substituí-los pela média ou mediana?
Obrigado.