setup.py

# Data setup
train_start_date = '2016-01-01'
train_end_date = '2020-06-30'

validate_start_date = '2020-07-01'
validate_end_date = '2021-12-31'

test_start_date = '2024-05-13'
test_end_date = '2023-06-06'

# Tickers list
train_tickerslist = 'stockallshares'
test_tickerslist = 'stockallshares'

# Tickers that exist the whole period, to ensure date consistency
# train_ticker_trading_base = 'AAPL' # sp500
train_ticker_trading_base = 'ABB.ST' # stockallshares

validate_ticker_trading_base = 'AAPL' # sp500
# validate_ticker_trading_base = 'ABB.ST' # stockallshares

# test_ticker_trading_base = 'ADDV-A.ST' # firstnorth
# test_ticker_trading_base = 'HEBA-B.ST' # omxmid
test_ticker_trading_base = 'ABB.ST' # omxlarge / stockallshares

data_interval = '1d'
data_groupby = 5
ma_period = 5

# Training model params
pretrained_model_name = 'tf_efficientnet_b7_ns'
max_epochs = 100
batch_size = 32
learning_rate = 0.001

# Test model
test_model_name = 'model_oburmo__actep6_apr06-0435_stockallshares_1d_tf_efficientnet_b7_ns_maxep100_bs32_lr0001'