ValueError: если n_samples=0, test_size=0,2 и train_size=None, результирующий набор поездов будет пустым. Отрегулируйте ⇐ Python

[Anonymous]

Я написал программу классификации текста. Когда я запускаю программу, она вылетает с ошибкой, как показано на этом снимке экрана:
[img]https://i.sstatic.net /wDoXv.jpg[/img]


ValueError: если n_samples=0, test_size=0.2 и train_size=None, результирующий набор поездов будет пустым. Настройте любой из вышеупомянутых параметров.

Вот мой код:

Код: Выделить всё

from sklearn.model_selection import train_test_split
from gensim.models.word2vec import Word2Vec
from sklearn.preprocessing import scale
from sklearn.linear_model import SGDClassifier
import nltk, string, json
import numpy as np

def cleanText(corpus):
reviews = []
for dd in corpus:
#for d in dd:
try:
words = nltk.word_tokenize(dd['description'])
words = [w.lower() for w in words]
reviews.append(words)
#break
except:
pass
return reviews

with open('C:\\NLP\\bad.json') as fin:
text = json.load(fin)
neg_rev = cleanText(text)

with open('C:\\NLP\\good.json') as fin:
text = json.load(fin)
pos_rev = cleanText(text)

#1 for positive sentiment, 0 for negative
y = np.concatenate((np.ones(len(pos_rev)), np.zeros(len(neg_rev))))

x_train, x_test, y_train, y_test = train_test_split(np.concatenate((pos_rev, neg_rev)), y, test_size=0.2)

Данные, которые я использую, доступны здесь:

• Плохо;
• Хорошо

Как бы я это исправил ошибка?

Подробнее здесь: https://stackoverflow.com/questions/600 ... esulting-t