Marshall-mk
/
Fetal-Health


  
1

	
2

	
3

	
4

	
5

	
6

	
7

	
8

	
9

	
10

	
11

	
12

	
13

	
14

	
15

	
16

	
17

	
18

	
19

	
20

	
21

	
22

	
23

	
24

	
25

	
26

	
27

	
28

	
29

	
30

	
31

	
32

	
33

	
34

	
35

	
36

	
37

	
38

	
39

	
40

	
41

	
42

	
43

	
44

	
45

	
            # train
cv: 3
estimator_name: xgb
estimators:
  knn: 
    param_grid: # params of GridSearchCV constructor
      n_neighbors: [4, 5]
      weights: ['uniform', 'distance']
      algorithm: 'auto'
      leaf_size: 30
      p: 2
      metric: 'minkowski'
      metric_params: None
      n_jobs: None
  xgb: 
    param_grid:
      n_estimators: [20, 30]
      booster: ['gbtree', 'gblinear']
  random_forest:
    param_grid:
      n_estimators: [50, 100]
      criterion: ['gini', 'log_loss']
      max_depth: None
      min_samples_split: 2
      min_samples_leaf: 1,
      min_weight_fraction_leaf: 0.0
      max_features: 'sqrt'

  decision_tree:
    param_grid:
      criterion: ['gini', 'entropy']
      splitter: 'best'
      max_depth: None
      min_samples_split: 2
      min_samples_leaf: 1
      min_weight_fraction_leaf: 0.0
      max_features: None
      random_state: None
      max_leaf_nodes: None
      min_impurity_decrease: 0.0
      class_weight: None
      ccp_alpha: 0.0

model_path: C:/Users/HI/Desktop/.dev/python/Deep learning/Projects/Fetal Health/models/model.joblib