【发布时间】:2017-05-08 12:18:17
【问题描述】:
我正在尝试此链接 http://deeplearning.net/tutorial/lstm.html 中的代码,并将 imdb 数据更改为我自己的数据。我正在尝试将此模型扩展为添加了“中性”类的 3 类多项式模型,而不是教程中提供的仅分类“正”和“负”类的二项式模型。
为了构建这个新模型,我重写了 build_dict 和 grab_data 函数来构建我的自定义字典并返回三个类的数据,并对 imdb_preprocess.py 中的>main():
dictionary = build_dict()
train_x_pos, train_x_neu, train_x_neg = grab_data('train', dictionary)
test_x_pos, test_x_neu, test_x_neg = grab_data('test', dictionary)
train_x = train_x_pos + train_x_neu + train_x_neg
train_y = [1] * len(train_x_pos) + [0] * len(train_x_neu) + [-1] * len(train_x_neg)
test_x = test_x_pos + test_x_neu + test_x_neg
test_y = [1] * len(test_x_pos) + [0] * len(test_x_neu) + [-1] * len(test_x_neg)
我对 train_y 和 test_y 的修改是否正确?
假设它们是正确的,我继续在我的数据上训练了这个模型,并且我成功地保存了 lstm_model.npz 和 lstm_model.npz.pkl 的新版本强>.
现在,如何在新示例中测试此模型?假设一个新的、看不见的例子是以下文本:
new_example = "I am very happy that I've been able to create a new LSTM sentiment model!"
我的理解是需要调用pred_probs函数,它接受参数f_pred_prob、prepare_data、data >、迭代器和verbose=False。我应该为这些参数传递什么?我猜函数调用会是这样的:
pred_probs(f_pred_prob, prepare_data, data=new_example, iterator, verbose=False)
【问题讨论】:
标签: deep-learning sentiment-analysis lstm