Adding check for nan values (#127)

Co-authored-by: Dmitry Razdoburdin

Author: razdoburdin

modelbuilders_bench/lgbm_mb.py

@@ -16,6 +16,8 @@
 
 import argparse
 import os
+import logging
+import sys
 
 import bench
 import daal4py
@@ -84,6 +86,10 @@
     '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,6 +15,8 @@
 # ===============================================================================
 
 import argparse
+import logging
+import sys
 
 import bench
 import daal4py
@@ -115,6 +117,10 @@ 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})