Functions — github.com/dawnranger/IDEC-pytorch

CodeHub MCPcopy Create free account

hub / github.com/dawnranger/IDEC-pytorch / functions

Functions13 in github.com/dawnranger/IDEC-pytorch

⨍ Functions 13 ◇ Types & classes 3

↓ 1 callersMethod__init__

(self, n_enc_1, n_enc_2, n_enc_3, n_dec_1, n_dec_2, n_dec_3, n_input, n_z)

↓ 1 callersFunctioncluster_acc

Calculate clustering accuracy. Require scikit-learn installed # Arguments y: true labels, numpy.array with shape `(n_samples,)`

↓ 1 callersFunctionload_mnist

(path='./data/mnist.npz')

↓ 1 callersMethodpretrain

(self, path='')

↓ 1 callersFunctionpretrain_ae

pretrain autoencoder

↓ 1 callersFunctiontarget_distribution

↓ 1 callersFunctiontrain_idec

Method__getitem__

(self, n_enc_1, n_enc_2, n_enc_3, n_dec_1,