search

indiejoseph / cnn-text-classification-tf-chinese

CNN for Chinese Text Classification in Tensorflow
235 stars 108 forks source link

请问想做多分类该怎么修改呢？ #15

Open YCG09 opened 7 years ago

YCG09 commented 7 years ago

你好，

我最近在做中文文本多分类，tensorflow刚开始学，还不太懂，如果我没搞错这个是做二分类的，请问要做多分类的话怎么修改呢？谢谢！

acalanatha commented 7 years ago

修改一下data_helpers.py中的函数，解析数据的label时支持一下多分类。 目前的label在内部表示是个长度为2的向量，如[0, 1] 如果是n分类的话，改成长度为n的向量，one_hot方式