"""Shared constants (values identical to ppi_benchmark_controls.py v1.0.0)."""

ENGINE_VERSION = "0.1.0"
FRAMEWORK_VERSION = "1.0.0"  # ppi_benchmark_controls.py release the algorithms come from
SCHEMA_VERSION = 1

AMINO_ACIDS = "ACDEFGHIKLMNPQRSTVWY"
PLM_MAX_RESIDUES = 1022

NEG_LABELS = {
    "N0": "Original negatives",
    "N1": "Random unobserved pairs",
    "N2": "Degree-matched",
    "N3": "Length-matched",
    "N5": "Degree + length matched",
    "N2-exact": "Exact-degree matched",
    "N2-nearest": "Nearest-degree matched",
    "N2-species": "Degree-matched within species",
}
REGIME_LABELS = {
    "R0": "Random pairs",
    "R1": "Both proteins seen",
    "R2": "One protein unseen",
    "R3": "Both proteins unseen",
    "R3-C": "Cluster-disjoint",
}
MODEL_LABELS = {"lr": "Logistic regression", "rf": "Random forest", "xgb": "XGBoost", "mlp": "MLP"}