dqn_noniid_4_20_difference.json

{
    "clients": {
        "total": 20,
        "per_round": 4,
        "label_distribution": "uniform"
    },
    "data": {
        "loading": "static",
        "partition": {
            "size": 3000
        },
        "IID": false,
        "bias": {
            "primary": 0.8,
            "secondary": false
        }
    },
    "federated_learning": {
        "rounds": 250,
        "target_accuracy": 0.99,
        "task": "train",
        "epochs": 5,
        "batch_size": 10
    },
    "model": "MNIST",
    "paths": {
        "data": "./data",
        "model": "./models",
        "case_name": "4-20_difference"        
    },
    "server": "dqn",
    "dqn": {
        "episode": 200,
        "max_steps": 50,
        "learning_rate": 0.0001,
        "gamma": 0.95,
        "epsilon_initial": 1.0,
        "epsilon_decay": 0.95,
        "epsilon_min": 0.01,
        "memory_size": 200,
        "batch_size": 4,
        "target_update": 10,
        "hidden_layers": [512, 512],
        "reward_fun": "difference",
        "reward_xi": 64,
        "rewards_log": "output/dqn_4-20_difference_rewards.csv",
        "saved_model": "output/dqn_models/dqn_4-20_difference_model",
        "trained_model": "output/dqn_models/dqn_4-20_difference_model_159.h5",
        "pca_model": "output/dqn_models/pca_4-20_difference.pkl"
    }
}