Как экспортировать данные обучения TensorFlow в CSV - PullRequest
1 голос
/ 13 июля 2020

Я запускаю проект анализа настроений SVM, и я хотел векторизовать его с помощью tf-idf. фрагмент кода для раздела tf-idf приведен ниже:

X = table_tfidf

y = df['label']

random_state = 42

from sklearn.model_selection import train_test_split

X_train, X_test, y_train, y_text = train_test_split(X, y, test_size=0.2, random_state=random_state)

#dividing training data to train data and validation data

X_train_train, X_train_val, y_train_train, y_train_val = train_test_split(X_train, y_train, test_size = 0.2)

как мне экспортировать данные X_train и только y_train в файл .csv? Я хотел проверить, правильно ли были разделены данные. Спасибо!

1 Ответ

1 голос
/ 13 июля 2020

Это должно сработать:

X_train.to_csv('X_train.csv')
...