Add sizes selection to runner py (#103)

* sizes in configs

* test config

* knn_svm

* Revert "knn_svm"

This reverts commit 6272ea5.

* revert changing skl config

* sizes for df_clsf

* update help info

Author: dmitrii-kriukov

configs/xpu/df_clsf.json

@@ -26,6 +26,7 @@
                     }
                 }
             ],
+            "workload-size": "medium",
             "num-trees": 10,
             "max-depth": 5
         },
@@ -46,6 +47,7 @@
                     }
                 }
             ],
+            "workload-size": "large",
             "num-trees": 100,
             "max-depth": 8
         },
@@ -66,6 +68,7 @@
                     }
                 }
             ],
+            "workload-size": "medium",
             "num-trees": 20,
             "max-depth": 16
         },
@@ -86,6 +89,7 @@
                     }
                 }
             ],
+            "workload-size": "large",
             "num-trees": 100,
             "max-depth": 10
         },
@@ -106,6 +110,7 @@
                     }
                 }
             ],
+            "workload-size": "medium",
             "num-trees": 50,
             "max-depth": 15
         }

runner.py

@@ -58,6 +58,11 @@ def get_configs(path: Path) -> List[str]:
                         help='Available floating point data types'
                         'This parameter only marks dtype as available, '
                         'make sure to add the dtype parameter to the config file ')
+    parser.add_argument('--workload-size', type=str, default="small medium large", nargs='+',
+                        choices=("small", "medium", "large"),
+                        help='Available workload sizes,'
+                        'make sure to add the workload-size parameter to the config file '
+                        'unmarked workloads will be launched anyway')
     parser.add_argument('--no-intel-optimized', default=False, action='store_true',
                         help='Use Scikit-learn without Intel optimizations')
     parser.add_argument('--output-file', default='results.json',
@@ -105,6 +110,11 @@ def get_configs(path: Path) -> List[str]:
             params = common_params.copy()
             params.update(params_set.copy())
 
+            if 'workload-size' in params:
+                if params['workload-size'] not in args.workload_size:
+                    continue
+                del params['workload-size']
+
             device = []
             if 'device' not in params:
                 if 'sklearn' in params['lib']: