From 571b18f87be9fbdba3f9ab52d3039b9e6c7e609d Mon Sep 17 00:00:00 2001 From: KOSASIH Date: Mon, 12 Aug 2024 11:58:43 +0700 Subject: [PATCH] Create config.py --- .../pi_network/pibrain/config.py | 34 +++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 blockchain_integration/pi_network/pibrain/config.py diff --git a/blockchain_integration/pi_network/pibrain/config.py b/blockchain_integration/pi_network/pibrain/config.py new file mode 100644 index 000000000..a71a792fe --- /dev/null +++ b/blockchain_integration/pi_network/pibrain/config.py @@ -0,0 +1,34 @@ +# config.py + +class Config: + """Configuration class for PiBrain.""" + + # General settings + PROJECT_NAME = "PiBrain" + VERSION = "1.0.0" + + # Data settings + DATA_DIR = "data" + MODEL_DIR = "models" + LOG_DIR = "logs" + + # Model settings + MODEL_TYPE = "neural_network" + HIDDEN_LAYERS = 2 + HIDDEN_UNITS = 128 + OUTPUT_UNITS = 10 + ACTIVATION_FUNCTION = "relu" + OPTIMIZER = "adam" + LOSS_FUNCTION = "mean_squared_error" + + # Training settings + BATCH_SIZE = 32 + EPOCHS = 10 + LEARNING_RATE = 0.001 + + # Evaluation settings + EVAL_METRICS = ["accuracy", "precision", "recall", "f1_score"] + + # Other settings + RANDOM_SEED = 42 + VERBOSE = True