-
Notifications
You must be signed in to change notification settings - Fork 2
Open
Description
Hi,
Thank you for this wonderful tool.
I've been unable to change the minimum samples for leaf nodes. Everything else works fine when generating the trees (I was able to change the maximum depth for example).
This is how I instantiated the tree:
set_tree_model = settree.SetTree(
classifier=False,
criterion="mse",
splitter='sklearn',
max_features=None,
min_samples_split=3,
operations=list_of_operations,
use_attention_set=USE_ATTN_SET,
use_attention_set_comp=USE_ATTN_SET_COMP,
attention_set_limit=ATTN_SET_LIMIT,
max_depth=2,
min_samples_leaf=3,
random_state=SEED)
Thanks again!
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels