Skip to content

Commit f6d2992

Browse files
committed
add note for VaDE
1 parent a86050e commit f6d2992

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

README.md

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,4 +13,10 @@ Implementation of Variational Deep Embedding from the IJCAI2017 paper:
1313

1414
Jiang, Zhuxi, et al. "Variational deep embedding: An unsupervised and generative approach to clustering." International Joint Conference on Artificial Intelligence. 2017.
1515

16-
The original code is written in [Keras](https://github.com/slim1017/VaDE). However, the original code is incorrect when computing the loss function. And I have corrected the loss function part with my code. The example usage can be found in `test/test_vade-3layer.py`, and it uses the pretrained weights from autoencoder in `test/model/pretrained_vade-3layer.pt`. Note: the pretrained weights is important to initialize the weights of VaDE.
16+
The original code is written in [Keras](https://github.com/slim1017/VaDE). However, the original code is incorrect when computing the loss function. And I have corrected the loss function part with my code. The example usage can be found in `test/test_vade-3layer.py`, and it uses the pretrained weights from autoencoder in `test/model/pretrained_vade-3layer.pt`.
17+
18+
Note:
19+
20+
* The pretrained weights is important to initialize the weights of VaDE.
21+
* Unlike the original code using combined training and test data for training and evaluation, I split the training and test data, and only use training data for training and test data for evaluation. I think it is a more appropriate way to evaluate the method for generalization.
22+
* As found, with above evaluation scheme and training for 3000 epochs, the clustering accuracy achieved is 93.65\%.

0 commit comments

Comments
 (0)