IntelPython / scikit-learn_bench

scikit-learn_bench benchmarks various implementations of machine learning algorithms across data analytics frameworks. It currently support the scikit-learn, DAAL4PY, cuML, and XGBoost frameworks for commonly used machine learning algorithms.
Apache License 2.0
111 stars 69 forks source link

Fix problem with float32 json dump #112

Closed dmitrii-kriukov closed 2 years ago

dmitrii-kriukov commented 2 years ago

json encoding error occured in pca, kmeans, dbscan with float32 when trying to json.dumps list congaing np.float32 now convert all np.float32 metrics to np.float64