How do i map subcellular localizations to lower the number of classes to predict
One way to do this would be to plot the first 10 GO categories of the the CC sub ontology (CC from cellular component). Or you can just visit the official webpage to see the list of children terms of them.

However there aren't only 4 or 5 categories, so you could arbitrary pick some of those.