sklearn多类分类器预测类的数量

Question

我正在研究一个多类分类问题，我需要在预测集中获得预定义数量的类。在下面的代码中，我使用sklearn的OnveVsRestClassifier进行Logistic回归

classifier = Pipeline([
        ('vectorizer', CountVectorizer(stop_words='english')),
        ('tfidf', TfidfTransformer()),
        ('clf', OneVsRestClassifier(LogisticRegression()))])

classifier.fit(X_train, Y_train)
predicted = classifier.predict(X_test)

上面的代码工作正常，它为每个测试用例返回可变数量的类。我想知道如何指定它为每个测试集返回N个预测类。