方法及其参数并在类中使用字典

Question

class Trainer():
  def __init__(self,param):
        ..........
        optimizer_target = getattr(optim, param['optimizer'])(net.parameters(), lr = param['learning_rate'])

     def train(epoch):
        .......... 

     def test(self, epoch):
        ..........

  def objective(self,train):
     
     params = {}
     model = Trainer(params)
     accuracy = model.test(trial)
      
     return accuracy 

if __name__ == '__main__':
    study = optuna.create_study(direction="maximize")
    study.optimize(objective)

当我在 main() 中调用这个目标函数时，它给了我不同的错误。不管我这样调用这个函数，它都会给我错误

study.optimize(lambda trial: Trainer.objective(trial)) 

如果我在没有类 Trainer 的情况下设计它，它会给我错误。请建议如何调用目标类的字典和在main()中调用目标函数

当前错误是

study.optimize（lambda 试验：Trainer.objective（试验）） 类型错误：objective() 缺少 1 个必需的位置参数：'trial'

Answer 1

您应该实例化您的类，或者将目标定义为@classmethod。例如：

class Trainer():
     def __init__(self,param):
        ..........
        optimizer_target = getattr(optim, param['optimizer'])(net.parameters(), lr = param['learning_rate'])

     def train(epoch):
        .......... 

     def test(self, epoch):
        ..........
     @classmethod
     def objective(self,train):
        params = {}

if __name__ == '__main__':
    study = optuna.create_study(direction="maximize")
    study.optimize(objective)

请参阅 python 文档中的类方法 (https://docs.python.org/3/library/functions.html#classmethod)。