【发布时间】:2020-11-02 21:11:40
【问题描述】:
我正在运行一个 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 文件中?我想检查数据是否正确拆分。谢谢!
【问题讨论】:
标签: python python-3.x jupyter-notebook tf-idf tfidfvectorizer