断言所有值都是有限的#
- sklearn.utils.assert_all_finite(X, *, allow_nan=False, estimator_name=None, input_name='')[source]#
如果X包含NaN或无穷大,则抛出ValueError异常。
- 参数:
- X{ndarray, 稀疏矩阵}
输入数据。
- allow_nan布尔值,默认为False
如果为True,则当
X
包含NaN时不抛出错误。- estimator_name字符串,默认为None
用于构建错误消息的估计器名称。
- input_namestr,默认值=””
用于构建错误消息的数据名称。特别是,如果
input_name
为“X”,数据包含 NaN 值且 allow_nan 为 False,则错误消息将链接到插补器文档。
示例
>>> 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.