constants.py

class MethodTypes:
    col_name: str = "method_type"
    automl: str = "AutoML"
    tree: str = "Tree-based"
    foundational: str = "Foundational"
    finetuned: str = "Neural-network"
    baseline: str = "Baseline"
    other: str = "Other"



model_type_emoji = {
    MethodTypes.tree: "🌴",
    MethodTypes.foundational: "🧠",
    MethodTypes.finetuned: "🌐",
    MethodTypes.automl: "🤖",
    MethodTypes.baseline: "📏",
    MethodTypes.other: "❓",
}