新闻文本分类问题

项目

数据集

课程

比赛

模型库

活动

论坛

访问飞桨官网

项目

数据集

课程

比赛

模型库

活动

论坛

访问飞桨官网

nk_ganggang 发布于2019-11

一个使用CNN进行文本分类的例子，链接如下：

https://aistudio.baidu.com/bdvgpu/user/83109/185742/notebooks/185742.ipynb?redirects=1

在AI Studio上可以正常运行

但是在本地运行时，训练、存储模型等都可以成功，但是在预测模块出现以下问题

PaddleCheckError: Tensor holds the wrong type, it holds int, but desires to be int64_t at [D:\1.6.1\paddle\paddle/fluid/framework/tensor_impl.h:30]

我直接用load_inference_model读取的模型，为什么跟存储的模型不一样。。。

[name: "save_infer_model/scale_0"
type {
type: LOD_TENSOR
lod_tensor {
tensor {
data_type: FP32
dims: -1
dims: 10
}
lod_level: 0
}
}

这里的int64为什么是FP32？

全部评论(4)

8melody8

#2 回复于2019-11

我也遇到了同样的问题，不知道如何解决

8melody8

#3 回复于2019-11

问题解决了，get_data()函数的返回值要做一下处理

def get_data(sentence):
    with open(dict_path, 'r', encoding='utf-8') as f_data:
        dict_txt = eval(f_data.readlines()[0])
    
    dict_txt = dict(dict_txt)
    
    keys = dict_txt.keys()
    
    data = []
    
    for s in sentence:
        if not s in keys:
            s = ''
        
        data.append(int(dict_txt[s]))
    
    return np.array(data, dtype = np.int64)

nk_ganggang

#4 回复于2019-11

8melody8 #3

问题解决了，get_data()函数的返回值要做一下处理 [代码]

真是太感谢了，我马上去试试~~~~

GoodmanWZ

#5 回复于2019-12

请问是怎么解决的！求告知

提issue

需求/bug反馈？一键提issue告诉我们

提pr

发现bug？如果您知道修复办法，欢迎提pr直接参与建设飞桨~