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