fixed bug with JSON parsing

Author: itearsl

bench.py

@@ -503,11 +503,10 @@ def gen_basic_dict(library, algorithm, stage, params, data, alg_instance=None,
             alg_instance_params = dict(alg_instance.attributes())
         else:
             alg_instance_params = dict(alg_instance.get_params())
-            # if ('min_samples_split' in alg_instance_params
-            #         and 'handle' in alg_instance_params
-            #         and 'cuml' in str(alg_instance_params['handle'])):
-            #     print(str(alg_instance_params['dtype']))
-            #     alg_instance_params['dtype'] = str(alg_instance_params['dtype'])
+            if ('min_samples_split' in alg_instance_params
+                    and 'handle' in alg_instance_params
+                    and 'cuml' in str(alg_instance_params['handle'])):
+                alg_instance_params['dtype'] = str(alg_instance_params['dtype'])
         result['algorithm_parameters'].update(alg_instance_params)
     if alg_params is not None:
         result['algorithm_parameters'].update(alg_params)

configs/testing/sklearn.json

@@ -244,7 +244,7 @@
                     "n_features": 10,
                     "training": {
                         "n_samples": 1000
-                    }                
+                    }
                 }
             ],
             "include-y": "",
@@ -323,6 +323,23 @@
                     }
                 }
             ]
+        },
+        {
+            "algorithm": "tsne",
+            "dataset": [
+                {
+                    "source": "synthetic",
+                    "type": "classification",
+                    "n_classes": 5,
+                    "n_features": 10,
+                    "training": {
+                        "n_samples": 1000
+                    },
+                    "testing": {
+                        "n_samples": 20
+                    }
+                }
+            ]
         }
     ]
 }