assert_all_finite

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

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

参数:

X{ndarray, 稀疏矩阵}

输入数据。

allow_nan布尔值, 默认为 False

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

estimator_name字符串, 默认为 None

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

input_name字符串, 默认为 “”

用于构建错误消息的数据名称。特别是如果 input_name 是“X”且数据包含 NaN 值且 allow_nan 为 False，则错误消息将链接到插补器（imputer）文档。

示例

>>> 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("Test passed: Array contains only finite values.")
... except ValueError:
...     print("Test failed: Array contains non-finite values.")
Test failed: Array contains non-finite values.