assert_all_finite

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

如果 X 包含 NaN 或無限值，則拋出 ValueError。

參數:

X{ndarray, 稀疏矩陣}

輸入資料。

allow_nanbool, default=False

如果為 True，當 X 包含 NaN 時，不會拋出錯誤。

estimator_namestr, default=None

估計器名稱，用於建構錯誤訊息。

input_namestr, default=””

用於建構錯誤訊息的資料名稱。特別是如果 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.