我有枚举python文件,其中包含:
class ClassificationType(object):
CLASSIFICATION_TYPE_UNSPECIFIED = 0
MULTICLASS = 1
MULTILABEL = 2
我正在编写另一个python文件来获取在枚举类中声明的变量的值。
def dataset(model_typ):
dataset_spec = {
"classification": enums.ClassificationType.MULTICLASS
}
按照上面的代码,我能够将MULTICLASS的值设为1。 现在我需要传递MULTICLASS / MULTILABEL / CLASSIFICATION_TYPE_UNSPECIFIED作为参数(model_type)并将其传递给dataset_spec。 怎么做?
提前致谢
注意:我不想更改enums.py文件。
答案 0 :(得分:0)
import { gql } from 'react-apollo';
答案 1 :(得分:0)
尝试在其他文件(不是enums.py文件)中使用以下代码:
from enums import ClassificationType as ct
import random
def dataset(model_typ):
dataset_spec = {
"classification": model_typ
}
print(dataset_spec)
dataset(random.choice([ct.MULTILABEL,ct.MULTICLASS]))
输出:
{'classification': 2}
我只是简单地将dataset_spec
' classification
键的值更改为参数model_typ
,然后在代码调用结束时调用dataset
函数并在参数中写入enums.ClassificationType.MULTICLASS
以获取enums.py
个文件MULTICLASS
变量