描述

我正在尝试实现一个仅包含几个参数的简单数据类。

@dataclass
class ReconstructionParameters:

    img_size: int
    CR: int
    denoise: bool
    epochs: int
    learning_rate: float
    step_size: int
    gamma: float
    batch_size: int
    regularization: float
    N0: float
    sig: float

    arch_name: InitVar[str]

    net_arch: int = field(init=False)


    def __post_init__(self, arch_name):
        self.net_arch = arch_name

    @property
    def net_arch(self):
        return str(netType(self.net_arch))

    @net_arch.setter
    def net_arch(self, arch_name):
        self.net_arch = int(netType[arch_name])
    

用户应该能够传递包含字符串的字符串arch_name在类的初始化期间，但类应保留以下定义的等效整数IntEnum:

class netType(IntEnum):
    
    c0mp = 0
    comp = 1
    pinv = 2
    free = 3

但是，如果用户想要获取先前创建的类中包含的 net_arch，他们应该有权访问用于初始化的相同字符串，而不是整数表示形式。我正在尝试使用@property装饰器和一个setter。理想情况下，我想使用__post_init__()初始化方法net_arch使用其设置器。

Error

运行以下代码时出现错误：

a = ReconstructionParameters(
    img_size=64, 
    CR=1024,
    denoise=True,
    epochs=20, 
    learning_rate=1e-6,
    step_size=10,
    gamma=1e-7,
    batch_size=256,
    regularization=1e-6, 
    N0=2500,
    sig=0.5,
    arch_name='c0mp')

Traceback (most recent call last):

  File "C:\spas\Programs\Python\test.py", line 51, in <module>
    a = ReconstructionParameters(

  File "<string>", line 14, in __init__

  File "C:\spas\Programs\Python\test.py", line 48, in net_arch
    self.net_arch = int(netType[arch_name])

  File "C:\Users\user\Anaconda3\envs\singlepixelenv\lib\enum.py", line 349, in __getitem__
    return cls._member_map_[name]

KeyError: <property object at 0x000001CF7C26BA40>

数据类字段和属性不能具有相同的名称。但是您可以在该字段中添加前导下划线，然后该属性将起作用。

from dataclasses import InitVar, dataclass, field
from enum import IntEnum


@dataclass
class ReconstructionParameters:
    img_size: int
    CR: int
    denoise: bool
    epochs: int
    learning_rate: float
    step_size: int
    gamma: float
    batch_size: int
    regularization: float
    N0: float
    sig: float

    arch_name: InitVar[str]

    _net_arch: int = field(init=False)

    def __post_init__(self, arch_name):
        self.net_arch = arch_name

    @property
    def net_arch(self):
        return str(netType(self._net_arch))

    @net_arch.setter
    def net_arch(self, arch_name):
        self._net_arch = int(netType[arch_name])


class netType(IntEnum):
    c0mp = 0
    comp = 1
    pinv = 2
    free = 3


a = ReconstructionParameters(
    img_size=64,
    CR=1024,
    denoise=True,
    epochs=20,
    learning_rate=1e-6,
    step_size=10,
    gamma=1e-7,
    batch_size=256,
    regularization=1e-6,
    N0=2500,
    sig=0.5,
    arch_name='c0mp')

print(a)
# ReconstructionParameters(img_size=64, CR=1024, denoise=True, epochs=20, learning_rate=1e-06, step_size=10, gamma=1e-07, batch_size=256, regularization=1e-06, N0=2500, sig=0.5, _net_arch=0)
print(a.net_arch)
# netType.c0mp