jnirschl
/
titanic_dvc
mirror of https://github.com/jnirschl/titanic_dvc.git


  
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

	
46

	
47

	
48

	
49

	
50

	
            classifier: random_forest
drop_cols:
- Name
- Cabin
- Ticket
dtypes:
  Age: float
  Embarked: category
  Fare: float
  Parch: int
  Pclass: category
  Sex: category
  SibSp: int
  Survived: category
feature_eng:
  featurize: false
imputation:
  Age: 29.6991
  Fare: 32.2042
  method: mean
model_params:
  logistic_regression: null
  naive_bayes: null
  neural_network: null
  random_forest:
    criterion: entropy
    max_depth: 20
    min_samples_leaf: 3
    min_samples_split: 9
  support_vector_machine: null
normalize: null
param_tuning:
  logistic_regression: null
  naive_bayes: null
  neural_network: null
  num_eval: 1000
  random_forest:
    criterion: entropy
    max_depth: 20
    min_samples_leaf: 3
    min_samples_split: 9
  scoring: roc_auc
  support_vector_machine: null
random_seed: 12345
train_test_split:
  n_split: 10
  shuffle: true
  target_class: Survived
predict:
  js_estimator: true