how to pre-train the model?

bdy9527 / SDCN

Structural Deep Clustering Network

Apache License 2.0

259 stars 72 forks source link

Closed dugzzuli closed 3 years ago

dugzzuli commented 4 years ago

I follow the parameters setting in the paper, to pre-train the model. However, the final result is not very good. Can you post the pre-training code?

bdy9527 commented 3 years ago

The pre-training code locates in SDCN/data/pretrain.py