Skip to content

Commit

Permalink
FIX add comment for future regression support
Browse files Browse the repository at this point in the history
  • Loading branch information
PSSF23 authored Dec 17, 2024
1 parent 2d9b6ea commit 58841ce
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion treeple/tree/_classes.py
Original file line number Diff line number Diff line change
Expand Up @@ -2441,7 +2441,7 @@ def __sklearn_tags__(self):
# XXX: nans should be supportable in SPORF by just using RF-like splits on missing values
# However, for MORF it is not supported
tags = super().__sklearn_tags__()
# tags.regressor_tags.multi_label = True
# tags.regressor_tags.multi_label = True TODO: add regression support
tags.input_tags.allow_nan = False
return tags

Expand Down

0 comments on commit 58841ce

Please sign in to comment.