pyhazards.configs package

Submodules

pyhazards.configs._schema module

class pyhazards.configs._schema.BenchmarkConfig(name, hazard_task, metrics=<factory>, eval_split='test', params=<factory>)

Bases: object

eval_split: str = 'test'

hazard_task: str

metrics: List[str]

name: str

params: Dict[str, Any]

class pyhazards.configs._schema.DatasetRef(name, params=<factory>)

Bases: object

name: str

params: Dict[str, Any]

class pyhazards.configs._schema.ExperimentConfig(benchmark, dataset, model, report=<factory>, seed=0, metadata=<factory>)

Bases: object

benchmark: BenchmarkConfig

dataset: DatasetRef

metadata: Dict[str, Any]

model: ModelRef

report: ReportConfig

seed: int = 0

to_dict()

Return type:

Dict[str, Any]

class pyhazards.configs._schema.ModelRef(name, task, params=<factory>)

Bases: object

name: str

params: Dict[str, Any]

task: str

class pyhazards.configs._schema.ReportConfig(output_dir='reports', formats=<factory>)

Bases: object

formats: List[str]

output_dir: str = 'reports'

pyhazards.configs._schema.dump_experiment_config(config, path)

Return type:

None

pyhazards.configs._schema.load_experiment_config(path)

Return type:

ExperimentConfig

Module contents

class pyhazards.configs.BenchmarkConfig(name, hazard_task, metrics=<factory>, eval_split='test', params=<factory>)

Bases: object

eval_split: str = 'test'

hazard_task: str

metrics: List[str]

name: str

params: Dict[str, Any]

class pyhazards.configs.DatasetRef(name, params=<factory>)

Bases: object

name: str

params: Dict[str, Any]

class pyhazards.configs.ExperimentConfig(benchmark, dataset, model, report=<factory>, seed=0, metadata=<factory>)

Bases: object

benchmark: BenchmarkConfig

dataset: DatasetRef

metadata: Dict[str, Any]

model: ModelRef

report: ReportConfig

seed: int = 0

to_dict()

Return type:

Dict[str, Any]

class pyhazards.configs.ModelRef(name, task, params=<factory>)

Bases: object

name: str

params: Dict[str, Any]

task: str

class pyhazards.configs.ReportConfig(output_dir='reports', formats=<factory>)

Bases: object

formats: List[str]

output_dir: str = 'reports'

pyhazards.configs.dump_experiment_config(config, path)

Return type:

None

pyhazards.configs.load_experiment_config(path)

Return type:

ExperimentConfig