Missing values support for inference (#136)

* np.float is depricated in numpy 1.24

* minor refactoring

* Removing check for nan

* remove unused imports

---------

Co-authored-by: Dmitry Razdoburdin <>

Author: razdoburdin

modelbuilders_bench/lgbm_mb.py

@@ -16,8 +16,6 @@
 
 import argparse
 import os
-import logging
-import sys
 
 import bench
 import daal4py
@@ -86,10 +84,6 @@
     'seed': params.seed
 }
 
-if np.isnan(X_test.values).any():
-    logging.warning('Nan values aren not supported in model builder yet')
-    sys.exit(1)
-
 if params.threads != -1:
     lgbm_params.update({'nthread': params.threads})
 

modelbuilders_bench/xgb_mb.py

@@ -15,8 +15,6 @@
 # ===============================================================================
 
 import argparse
-import logging
-import sys
 
 import bench
 import daal4py
@@ -117,10 +115,6 @@ def convert_xgb_predictions(y_pred, objective):
         params.enable_experimental_json_serialization
 }
 
-if np.isnan(X_test.values).any():
-    logging.warning('Nan values aren not supported in model builder yet')
-    sys.exit(1)
-
 if params.threads != -1:
     xgb_params.update({'nthread': params.threads})