金属列车收到了意外的关键字参数“ n_epochs”

Question

我正在编写Python代码，并且在代码的这一部分中，我想使用Metal来训练我的模型，就像这里：

from metal.label_model import LabelModel
gen_model = LabelModel()
%time gen_model.train(L_train[0], n_epochs=500, print_every=100)

但这给出了：

TypeError                                 
Traceback (most recent call last)
<timed eval> in <module>

TypeError: train() got an unexpected keyword argument 'n_epochs'

Answer 1

在0.3.0中有一个更改：

  

'将Classifier.train重命名为Classifier.train_model以避免   覆盖nn.Module.train函数'

因此，请尝试使用train_model()代替train()：

from metal.label_model import LabelModel
gen_model = LabelModel()
%time gen_model.train_model(L_train[0], n_epochs=500, print_every=100)

来源：

https://github.com/HazyResearch/metal/commit/4210c7c66f3f4a6fc7287192aec133c293ed8198