How to do unsupervised pretraining in dl4j with autoencoder and then doing supervised training in order to classify the differentes labels ?
Thanks in advance !
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.
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.