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!
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!