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) ...@@ -232,7 +232,7 @@ opt_crc = tf.train.AdamOptimizer(learning_rate=0.0001).minimize(cost_crc)
tf.global_variables_initializer().run() tf.global_variables_initializer().run()
batchsz = 16 batchsz = 16
hm_epochs = 4#300 hm_epochs = 300
iterations = int(data.shape[0] / batchsz) iterations = int(data.shape[0] / batchsz)
max_length = data.shape[1] max_length = data.shape[1]
...@@ -243,7 +243,7 @@ if data.shape[0] % batchsz != 0: ...@@ -243,7 +243,7 @@ if data.shape[0] % batchsz != 0:
best_loss = sys.float_info.max best_loss = sys.float_info.max
noise_factor = 0.01 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 new_centroids = None
kmeans_labels = 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