Skip to content

Commit 0737115

Browse files
committed
Delete test codes
1 parent aab4242 commit 0737115

File tree

1 file changed

+0
-25
lines changed

1 file changed

+0
-25
lines changed

_Dist/NeuralNetworks/d_Traditional2NN/Toolbox.py

-25
Original file line numberDiff line numberDiff line change
@@ -172,28 +172,3 @@ def _transform(self):
172172
w2 *= max_route_length
173173
self._transform_ws = [w1, w2, w3]
174174
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

Comments
 (0)