将scipy.sparse矩阵拆分为训练和测试

时间:2018-06-23 16:00:46

标签: python pandas scipy sparse-matrix

sc_f = sc27.tocsr()
X, y = sc_f
X_train,X_test, y_train, y_test = train_test_split(sc_f, y, stratify=y, test_size=0.3)

将稀疏矩阵拆分为Train and Test时出现错误。 如何将阵列拆分为训练和测试?

sc_f:稀疏稀疏矩阵

我想以某种方式拆分它,以便在y中有我要预测的目标即“ is_true”,并且仅采用两个值0和1。 当我将scipy稀疏矩阵转换为sparseDataFrame时。我无法以df ['is_true']的身份访问它,其中df是sparseDataFrame。我如何分割它。有人可以帮忙吗?

0 个答案:

没有答案