Conditions | 1 |
Total Lines | 10 |
Code Lines | 8 |
Lines | 10 |
Ratio | 100 % |
Changes | 0 |
1 | # Author: Simon Blanke |
||
17 | View Code Duplication | def model(para, X_train, y_train): |
|
|
|||
18 | model = DecisionTreeClassifier( |
||
19 | criterion=para["criterion"], |
||
20 | max_depth=para["max_depth"], |
||
21 | min_samples_split=para["min_samples_split"], |
||
22 | min_samples_leaf=para["min_samples_leaf"], |
||
23 | ) |
||
24 | scores = cross_val_score(model, X_train, y_train, cv=2) |
||
25 | |||
26 | return scores.mean() |
||
27 | |||
47 |