python - 如何将 TensorFlow 训练数据导出为 CSV

Question

我正在运行一个 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 文件中？我想检查数据是否正确拆分。谢谢！

标签： pythonpython-3.xjupyter-notebooktf-idftfidfvectorizer

Answer 1

这应该这样做：

X_train.to_csv('X_train.csv')