Skip to content

Commit

Permalink
Create config.py
Browse files Browse the repository at this point in the history
  • Loading branch information
KOSASIH authored Aug 12, 2024
1 parent 885cea5 commit 571b18f
Showing 1 changed file with 34 additions and 0 deletions.
34 changes: 34 additions & 0 deletions blockchain_integration/pi_network/pibrain/config.py
Original file line number Diff line number Diff line change
@@ -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

0 comments on commit 571b18f

Please sign in to comment.