Refactor ComponentSelector
to match sklearn's DecisionTreeClassifier
#1034
Labels
refactoring
issues proposing/requesting changes to the code which do not impact behavior
TE-dependence
issues related to TE dependence metrics and component selection
Summary
If we use
DecisionTreeClassifier
as the general template forComponentSelector
.__init__()
takes hyperparameters, but not data. In our case, that would be thetree
, but not thecomptable
.fit()
takes training data and fits the model. In our case,comptable
is training data.predict()
takes test data and predicts labels. In our case,comptable
is the test data as well.If we organize the tree's attributes similarly to
DecisionTreeClassifier
, we might be able to directly use sklearn'splot_tree
.It would also be really cool if we could figure out how to plot the decision surfaces based on features (see this example).
The text was updated successfully, but these errors were encountered: