Skip to content

Unable to define minimum samples for leaf nodes #2

Description

@hravkin

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!

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions