Я работаю над классификацией текста с метками Multi class. В настоящее время я беру столбец Description
как независимую переменную. Я вычисляю Word2ve c для Description
столбца и принимаю его как X
и передаю метки y
. После создания вектора предложения я разделяю данные на «поезд» и «тест», как показано ниже:
training_data, testing_data, training_labels, testing_labels = \
train_test_split(combined_sentence_vectors, df.labels, test_size=0.1)
, а затем я использую OneVsRest Classifier
:
clf1 = OneVsRestClassifier(LinearSVC(class_weight='balanced',random_state=0)).fit(training_data, training_labels)
. Я хочу добавить еще одну функцию df['Status']
к своим данным «Обучение». Как я могу это сделать? Как я могу добавить эту функцию к моему вектору предложения или к моим «обучающим данным»?