Spaces:

sundea
/

text-classification

Runtime error

sundea commited on Jun 12, 2023

Commit

1015b9c

1 Parent(s): a745fd3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,8 +9,8 @@ import torch
 import pickle as pkl
 from utils import build_dataset
-classes = ['finance', 'realty', 'stocks', 'education', 'science', 'society', 'politics', 'sports', 'game',
-           'entertainment']
 MAX_VOCAB_SIZE = 10000  # 词表长度限制
 UNK, PAD = '<UNK>', '<PAD>'  # 未知字，padding符号

 import pickle as pkl
 from utils import build_dataset
+classes = ['金融类', '房地产类', '股票类', '教育类', '科技类', '社会类', '政治类', '体育类', '游戏类',
+           '娱乐类']
 MAX_VOCAB_SIZE = 10000  # 词表长度限制
 UNK, PAD = '<UNK>', '<PAD>'  # 未知字，padding符号