Supervised Text Classification Python