assert_all_finite

sklearn.utils.assert_all_finite(X, *, allow_nan=False, estimator_name=None, input_name='')

抛出ValueError如果X包含NaN或无穷大。

Parameters:

X{ndarray, sparse matrix}

输入数据。

allow_nanbool, default=False

如果为True，当 X 包含NaN时不抛出错误。

estimator_namestr, default=None

估计器名称，用于构造错误消息。

input_namestr, default=””

用于构造错误消息的数据名称。特别是如果 input_name 是”X”并且数据包含NaN值且allow_nan为False，错误消息将链接到imputer文档。

Examples

>>> from sklearn.utils import assert_all_finite
>>> import numpy as np
>>> array = np.array([1, np.inf, np.nan, 4])
>>> try:
...     assert_all_finite(array)
...     print("测试通过：数组仅包含有限值。")
... except ValueError:
...     print("测试失败：数组包含非有限值。")
测试失败：数组包含非有限值。