python - 将实例变量传递给装饰器
问题描述
我发现这个有用的装饰器允许你传入一些可选参数
def mlflow_experiment(
_func=None,
*,
experiment_name=None
):
def experiment_decorator(func):
@functools.wraps(func)
def experiment_wrapper(self, *args, **kwargs):
nonlocal experiment_name
experiment_id = (
mlflow.set_experiment(experiment_name)
if experiment_name is not None
else None
)
...
value = func(self, *args, **kwargs)
return value
return experiment_wrapper
if _func is None:
return experiment_decorator
else:
return experiment_decorator(_func)
因此,在这样的用例中,我只是将字符串传递给experiment_name
,代码可以完美运行。
@mlflow_experiment(autolog=True, experiment_name = 'blarg')
def train_mlflow(self, maxevals=50, model_id=0):
...
我一直很难弄清楚装饰器的作用域,但我并不感到惊讶,使用传递定义的实例变量__init__
不起作用。
class LGBOptimizerMlfow:
def __init__(self, arg):
self.arg = arg
@mlflow_experiment(autolog=True, experiment_name = self.arg)
def train_mlflow(self, maxevals=50, model_id=0):
...
>>> `NameError: name 'self' is not defined`
只是为了看看作用域是否是一个问题,我在类之外声明了变量并且它起作用了。
只是为了它,我决定在类中声明一个全局变量,它也可以工作,但它不太理想,特别是如果我想将它作为可选参数传递给类或方法。
class LGBOptimizerMlfow:
global arg
arg = 'hi'
@mlflow_experiment(autolog=True, experiment_name = arg)
def train_mlflow(self, maxevals=50, model_id=0):
...
修改代码以使装饰器接受实例变量的任何帮助都会很可爱。
谢谢!
解决方案
装饰器是在定义类时调用的,self
它只是用于每个实例方法的参数,而不是类本身提供的东西。所以self
在你需要它用作装饰器的参数时没有定义。
您需要修改experiment_wrapper
以直接从其 self
参数中获取名称,而不是从mflow_experiment
. 就像是
def mlflow_experiment(
_func=None,
*,
experiment_name=None,
tracking_uri=None,
autolog=False,
run_name=None,
tags=None,
):
def experiment_decorator(func):
@functools.wraps(func)
def experiment_wrapper(self, *args, **kwargs):
nonlocal tracking_uri
experiment_name = getattr(self, 'experiment_name', None)
experiment_id = (
mlflow.set_experiment(experiment_name)
if experiment_name is not None
else None
)
...
with mlflow.start_run(experiment_id=experiment_id
, run_name=run_name
, tags=tags):
value = func(self, *args, **kwargs)
return value
return experiment_wrapper
if _func is None:
return experiment_decorator
else:
return experiment_decorator(_func)
然后,您需要确保每个实例都有一个与之关联的实验名称(或无)。
class LGBOptimizerMlfow:
def __init__(self, arg, experiment_name=None):
self.arg = arg
self.experiment_name = experiment_name
@mlflow_experiment(autolog=True, experiment_name = self.arg)
def train_mlflow(self, maxevals=50, model_id=0):
...
另一种选择是对 进行experiment_name
参数train_mflow
,从而更容易使用相同的方法创建不同的名称。(这可能更接近您的意图。)
class LGBOptimizerMlfow:
def __init__(self, arg):
self.arg = arg
@mlflow_experiment(autolog=True)
def train_mlflow(self, maxevals=50, model_id=0, experiment_name=None):
if experiment_name is None:
self.experiment_name = self.arg
...
装饰器的定义和上面一样。
推荐阅读
- java - runnodes时发生了非法反射访问操作
- sql - SQL:如何减去两个字段属于同一个ID
- spring - 使用 Json Serializable 类将 base64 字符串转换为字节数组
- php - 是否可以从 OneDrive 文件夹中获取永久链接?
- java - 当应用程序进入后台时停止 MediaPlayer 服务
- reactjs - react-konva - 处理变压器选择,取消选择和悬停?
- python - Django - ModelForm 中的选择列表
- mysql - 想要更新 mysql 字段值,如 00001、00002、00003
- java - UcanAccess 和 Java 出现 BIG_INT 错误
- javascript - 在 url 更改时加载事件