config.json

{
    "model_name": "Magical-1 Star",
    "version": "1.0.0",
    "sentence_transformer": {
        "model_name": "all-MiniLM-L6-v2",
        "max_seq_length": 128
    },
    "classifier": {
        "input_size": 384,
        "num_classes": 2
    },
    "training": {
        "batch_size": 32,
        "num_epochs": 5,
        "learning_rate": 2e-5,
        "optimizer": "AdamW",
        "loss_function": "CrossEntropyLoss"
    },
    "data": {
        "train_test_split": 0.2,
        "random_seed": 42
    },
    "paths": {
        "model_save_path": "magical_1_star.safetensors",
        "log_dir": "logs"
    }
}