Металлический поезд получил неожиданный ключевой аргумент «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