src/constants.py

from dataclasses import dataclass


class MetricNames:
    raw_error: str = "raw-error"
    normalized_error: str = "normalized-error"
    fit_time_per_1K_rows: str = "fit-time-per-1K-rows"
    inference_time_per_1K_rows: str = "inference-time-per-1K-rows"
    rank: str = "rank"


class ProblemTypes:
    col_name: str = "problem_type"
    regression: str = "Regression"
    classification: str = "Classification"
    multi_classification: str = "Multi-classification"


class MethodTypes:
    col_name: str = "method_type"
    automl: str = "AutoML"
    tree: str = "Tree-based"
    foundational: str = "Foundational"
    finetuned: str = "Fine-tuned"
    other: str = "Other"

class DatasetInfo:
    col_name: str = "dataset"
    num_rows: str = "num_rows"
    num_features: str = "num_features"

@dataclass
class DatasetInfoRow:
    name: str
    problem_type: str
    num_rows: int
    num_features: int