|
62 | 62 | modes: unknown, classification, regression |
63 | 63 | |
64 | 64 | engines: |
65 | | - classification: brulee, keras, nnet |
66 | | - regression: brulee, keras, nnet |
| 65 | + classification: brulee, brulee_two_layer, keras, nnet |
| 66 | + regression: brulee, brulee_two_layer, keras, nnet |
67 | 67 | |
68 | 68 | |
69 | 69 | arguments: |
70 | | - keras: |
| 70 | + keras: |
71 | 71 | hidden_units --> hidden_units |
72 | 72 | penalty --> penalty |
73 | 73 | dropout --> dropout |
74 | 74 | epochs --> epochs |
75 | 75 | activation --> activation |
76 | | - nnet: |
| 76 | + nnet: |
77 | 77 | hidden_units --> size |
78 | 78 | penalty --> decay |
79 | 79 | epochs --> maxit |
80 | | - brulee: |
| 80 | + brulee: |
| 81 | + hidden_units --> hidden_units |
| 82 | + penalty --> penalty |
| 83 | + epochs --> epochs |
| 84 | + dropout --> dropout |
| 85 | + learn_rate --> learn_rate |
| 86 | + activation --> activation |
| 87 | + brulee_two_layer: |
81 | 88 | hidden_units --> hidden_units |
82 | 89 | penalty --> penalty |
83 | 90 | epochs --> epochs |
|
86 | 93 | activation --> activation |
87 | 94 | |
88 | 95 | fit modules: |
89 | | - engine mode |
90 | | - keras regression |
91 | | - keras classification |
92 | | - nnet regression |
93 | | - nnet classification |
94 | | - brulee regression |
95 | | - brulee classification |
| 96 | + engine mode |
| 97 | + keras regression |
| 98 | + keras classification |
| 99 | + nnet regression |
| 100 | + nnet classification |
| 101 | + brulee regression |
| 102 | + brulee classification |
| 103 | + brulee_two_layer regression |
| 104 | + brulee_two_layer classification |
96 | 105 | |
97 | 106 | prediction modules: |
98 | | - mode engine methods |
99 | | - classification brulee class, prob |
100 | | - classification keras class, prob, raw |
101 | | - classification nnet class, prob, raw |
102 | | - regression brulee numeric |
103 | | - regression keras numeric, raw |
104 | | - regression nnet numeric, raw |
| 107 | + mode engine methods |
| 108 | + classification brulee class, prob |
| 109 | + classification brulee_two_layer class, prob |
| 110 | + classification keras class, prob, raw |
| 111 | + classification nnet class, prob, raw |
| 112 | + regression brulee numeric |
| 113 | + regression brulee_two_layer numeric |
| 114 | + regression keras numeric, raw |
| 115 | + regression nnet numeric, raw |
105 | 116 | |
106 | 117 |
|
0 commit comments