Если вы проверяете dtype
вашего массива X_to_predict
, он должен показывать float64
.
# slightly modified array from the question
X_to_predict = np.array([1.37097033e+002, 0.00000000e+000, -1.82710826e+296,
1.22703799e+002, 1.37097033e+002, -2.56391552e+001,
1.11457878e+002, 1.37097033e+002, -2.56391552e+001,
9.81898928e+001, 1.22703799e+002, -2.45139066e+001]).reshape((3, 4))
print(X_to_predict.dtype)
>>> float64
RandomForestClassifier sklearn молча преобразует массив в float32
, см. Обсуждение здесь для источника сообщения об ошибке.
Вы можете преобразовать его самостоятельно
print(X_to_predict.astype(np.float32)))
>>> array([[137.09703 , 0. , -inf, 122.7038 ],
[137.09703 , -25.639154, 111.45788 , 137.09703 ],
[-25.639154, 98.189896, 122.7038 , -24.513906]],
dtype=float32)
Третье значение (-1.82710826e + 296) становится -inf
в float32.Единственный способ обойти это - заменить значения inf
на максимум с плавающей точкой32.Вы потеряете некоторую точность, насколько я знаю, в настоящее время нет параметров или обходного пути, за исключением изменения реализации в sklearn и его перекомпиляции.
Если вы используете np.nan_to_num
, ваш массив должен выглядеть следующим образом:
new_X = np.nan_to_num(X_to_predict.astype(np.float32))
print(new_X)
>>> array([[ 1.3709703e+02, 0.0000000e+00, -3.4028235e+38, 1.2270380e+02],
[ 1.3709703e+02, -2.5639154e+01, 1.1145788e+02, 1.3709703e+02],
[-2.5639154e+01, 9.8189896e+01, 1.2270380e+02, -2.4513906e+01]],
dtype=float32)
, который должен быть принят вашим классификатором.
Полный код
import numpy as np
from sklearn.ensemble import RandomForestClassifier
from sklearn.datasets import load_iris
iris = load_iris()
clf = RandomForestClassifier(n_estimators=10,
random_state=42)
clf.fit(iris.data, iris.target)
X_to_predict = np.array([1.37097033e+002, 0.00000000e+000, -1.82710826e+296,
1.22703799e+002, 1.37097033e+002, -2.56391552e+001,
1.11457878e+002, 1.37097033e+002, -2.56391552e+001,
9.81898928e+001, 1.22703799e+002, -2.45139066e+001]).reshape((3, 4))
print(X_to_predict.dtype)
print(X_to_predict.astype(np.float32))
new_X = np.nan_to_num(X_to_predict.astype(np.float32))
print(new_X)
#should return array([2, 2, 0])
print(clf.predict(new_X))
# should crash
clf.predict(X_to_predict)