We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent aab4242 commit 0737115Copy full SHA for 0737115
_Dist/NeuralNetworks/d_Traditional2NN/Toolbox.py
@@ -172,28 +172,3 @@ def _transform(self):
172
w2 *= max_route_length
173
self._transform_ws = [w1, w2, w3]
174
self._transform_bs = [b]
175
-
176
177
-if __name__ == '__main__':
178
- train, test = np.load("train.npy"), np.load("test.npy")
179
- x, y = train[..., :-1], train[..., -1]
180
- x_test, y_test = test[..., :-1], test[..., -1]
181
- # x_mean, x_std = x.mean(), x.std()
182
- # x -= x_mean; x /= x_std
183
- # x_test -= x_mean; x_test /= x_std
184
- from _Dist.NeuralNetworks.e_AdvancedNN.NN import Advanced
185
- Advanced(
186
- name="madelon",
187
- data_info={
188
- "numerical_idx": [True] * 500 + [False],
189
- "categorical_columns": []
190
- },
191
- model_param_settings={
192
- "lr": 1e-3,
193
- "activations": ["relu", "relu"]
194
- }, model_structure_settings={
195
- "use_pruner": False,
196
- "use_wide_network": False,
197
- "hidden_units": [152, 153]
198
- }
199
- ).fit(x, y, x_test, y_test, snapshot_ratio=1)
0 commit comments