nlpstack.tasks.classification.torch module#