def getTF(dataset): tfidf = TfidfTransformer(norm=None) tfidf.fit(dataset['train']) return tfidf.idf_