@ptrblck my classification task is binary 0 or 1

i want to reduce 118 features to some x number until data representation is not lost
Rather then feeding 118 features to any algorithm, i want to give some smaller features set which represent 118 original features, hence i started to reduce 118 features to 100 using autoencoder
Then these 100 features will predict classification 1 and 0 using any algo e.g. LGBM in my case

The classifier head here is 111 as i followed your given example , it is not 2 for 0 or 1 classification
self.decoder_classifier = nn.Linear(480,111)
Do you want me to change it
self.decoder_classifier = nn.Linear(480,2)