22 lines
719 B
Python
22 lines
719 B
Python
from dataclasses import dataclass
|
|
|
|
|
|
@dataclass
|
|
class DataConfig:
|
|
n_features: int
|
|
n_informative: int
|
|
n_redundant: int
|
|
n_clusters_per_class: int
|
|
flip_y:float
|
|
|
|
config_easy = DataConfig(n_features=2, n_informative=2, n_redundant=0, n_clusters_per_class=1, flip_y=0.0)
|
|
config_medium = DataConfig(n_features=10, n_informative=5, n_redundant=2, n_clusters_per_class=2, flip_y=0.01)
|
|
config_hard = DataConfig(n_features=50, n_informative=20, n_redundant=10, n_clusters_per_class=4, flip_y=0.05)
|
|
config_veryhard = DataConfig(n_features=100, n_informative=30, n_redundant=20, n_clusters_per_class=8, flip_y=0.05)
|
|
|
|
configs = [
|
|
config_easy,
|
|
config_medium,
|
|
config_hard,
|
|
config_veryhard
|
|
] |