DNN文本分类出现的问题

用DNN分类的时候出现了以下问题,麻烦大家帮忙解答一下:

history = model.fit(xtrain_count, ytrain_count,
                    batch_size=batch_size,
                    epochs=50,callbacks=[checkpoint],
                    verbose=1,
                    validation_split=0.1)

ValueError: `validation_split` is only supported for Tensors or NumPy arrays, found following types in the input: [<class 'scipy.sparse.csr.csr_matrix'>]

谢谢!

如果使用validation_split分割验证集,那么输入的训练集必须是Tensor或者Numpy数据类型。类似问题参考这里:
https://blog.csdn.net/m0_47256162/article/details/122309758

根据报错提示,最后一个参数写错了,0.1那个,不支持这种类型

您好,我是有问必答小助手,您的问题已经有小伙伴帮您解答,感谢您对有问必答的支持与关注!
PS:问答VIP年卡 【限时加赠:IT技术图书免费领】,了解详情>>> https://vip.csdn.net/askvip?utm_source=1146287632

原来是矩阵类型,转成tensor类型就好了
xtrain_count=scipy_sparse_mat_to_torch_sparse_tensor(xtrain_count)
ytrain_count=scipy_sparse_mat_to_torch_sparse_tensor(ytrain_count)