repo_id = "kaenova/headline_detector"

indobertweet_checkpoints = {
    1: "indobertweet/scenario_1_val_epoch=2-validation_loss=0.0542.ckpt",
    2: "indobertweet/scenario_2_val_epoch=1-validation_loss=0.0444.ckpt",
    3: "indobertweet/scenario_3_val_epoch=0-validation_loss=0.0793.ckpt",
    4: "indobertweet/scenario_4_val_epoch=1-validation_loss=0.1154.ckpt",
    5: "indobertweet/scenario_5_val_epoch=0-validation_loss=0.0691.ckpt",
    6: "indobertweet/scenario_6_val_epoch=1-validation_loss=0.1011.ckpt",
}

cnn_checkpoints = {
    1: "cnn/scenario_1_val_epoch=5-validation_loss=0.2260.ckpt",
    2: "cnn/scenario_2_val_epoch=3-validation_loss=0.2222.ckpt",
    3: "cnn/scenario_3_val_epoch=2-validation_loss=0.2587.ckpt",
    4: "cnn/scenario_4_val_epoch=1-validation_loss=0.2822.ckpt",
    5: "cnn/scenario_5_val_epoch=2-validation_loss=0.2576.ckpt",
    6: "cnn/scenario_6_val_epoch=1-validation_loss=0.2807.ckpt",
}

fasttext_checkpoints = {
    1: "fasttext/scenario_1_val_epoch=53-validation_loss=0.2726.ckpt",
    2: "fasttext/scenario_1_val_epoch=53-validation_loss=0.2726.ckpt",
    3: "fasttext/scenario_3_val_epoch=50-validation_loss=0.2775.ckpt",
    4: "fasttext/scenario_4_val_epoch=29-validation_loss=0.3066.ckpt",
    5: "fasttext/scenario_5_val_epoch=48-validation_loss=0.2798.ckpt",
    6: "fasttext/scenario_6_val_epoch=28-validation_loss=0.3102.ckpt",
}