Unsupervised pretraining with autoencoder and supervised training

How to do unsupervised pretraining in dl4j with autoencoder and then doing supervised training in order to classify the differentes labels ?

Thanks in advance !

Aside from calling fit to train on labeled data, you can also call pretrain to pretrain your model in an unsupervised fashion.

1 Like

That helps as well !
But to answer to my asked question, I did transferLearning to take pretrained model in autoencoder(model.fit(input, input)) and apply dense neural network with activation function softmax in output layer and doing supervised training.