Open w-wenjie opened 5 years ago
optimize_class_model方法里调用create_classification_model时会传入num_labels,但是这个num_labels在前面没定义。 #############################################################
if not args.num_labels: num_labels, label2id, id2label = init_predict_var(tmp_dir)
############################################################# 如果执行脚本时输入了num_labels参数,则上面这段代码就不会执行,这就导致没有定义num_labels变量。
将183行代码的参数num_labels=num_labels改为num_labels=args.num_labels即可。
optimize_class_model方法里调用create_classification_model时会传入num_labels,但是这个num_labels在前面没定义。 #############################################################
增加 从label2id.pkl中读取num_labels, 这样也可以不用指定num_labels参数; 2019/4/17
############################################################# 如果执行脚本时输入了num_labels参数,则上面这段代码就不会执行,这就导致没有定义num_labels变量。