Commit 63366790 authored by Ienco Dino's avatar Ienco Dino
Browse files

Update DETSEC.py

parent de185f5e
......@@ -232,7 +232,7 @@ opt_crc = tf.train.AdamOptimizer(learning_rate=0.0001).minimize(cost_crc)
tf.global_variables_initializer().run()
batchsz = 16
hm_epochs = 4#300
hm_epochs = 300
iterations = int(data.shape[0] / batchsz)
max_length = data.shape[1]
......@@ -243,7 +243,7 @@ if data.shape[0] % batchsz != 0:
best_loss = sys.float_info.max
noise_factor = 0.01
th = 2#number of epochs for the autoencoder pretraining step
th = 50#number of epochs for the autoencoder pretraining step
new_centroids = None
kmeans_labels = None
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment