You can specify the classes to train using the classes argument.
classes
model.train(classes=[0,1,2,3,....])
credits: YTG from github