feat: feat_select (#38)

* feat_select * fix * new_type
ntua-unit-of-control-and-informatics · Oct 30, 2024 · 19cbebb · 19cbebb
1 parent e6e34f4
commit 19cbebb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/helpers/json_to_predreq.py b/src/helpers/json_to_predreq.py
@@ -36,4 +36,6 @@ def decode(request):
         task=request.model.task.lower(),
         featurizer=featurizers,
     )
+    if len(request.model.selected_features) > 0:
+        dataset.select_features(SelectionList=request.model.selected_features)
     return dataset, jaqpot_row_ids