from utils import read_stock_market_data
from evaluating.portfolio_evaluator import PortfolioEvaluator
from model.CompanyEnum import CompanyEnum
from dependency_injection_containers import Traders
import datetime
from definitions import PERIOD_1, PERIOD_2
if __name__ == "__main__":
# Load stock market data for training and testing period
stock_market_data = read_stock_market_data(
[CompanyEnum.COMPANY_A, CompanyEnum.COMPANY_B], [PERIOD_1, PERIOD_2])
# Define portfolio-name/trader mappings
portfolio_name_trader_mappings = [
# Benchmark trader
('Buy-and-hold Trader', Traders.BuyAndHoldTrader()),
# Simple traders
('Simple Trader (perfect prediction)',
Traders.SimpleTrader_with_perfect_prediction()),
('Simple Trader (NN binary perfect prediction)',
Traders.SimpleTrader_with_nn_binary_perfect_prediction()),
('Simple Trader (NN binary prediction)',
Traders.SimpleTrader_with_nn_binary_prediction()),
# Deep Q-Learning traders
('DQL Trader (perfect prediction)',
Traders.DqlTrader_with_perfect_prediction()),
('DQL Trader (NN binary perfect prediction)',
Traders.DqlTrader_with_nn_binary_perfect_prediction()),
('DQL Trader (NN binary prediction)',
from utils import read_stock_market_data
from evaluating.portfolio_evaluator import PortfolioEvaluator
from model.CompanyEnum import CompanyEnum
from dependency_injection_containers import Traders
import datetime
from definitions import PERIOD_1, PERIOD_2
if __name__ == "__main__":
# Load stock market data for training and testing period
stock_market_data = read_stock_market_data(
[CompanyEnum.COMPANY_A, CompanyEnum.COMPANY_B], [PERIOD_1, PERIOD_2])
# Define portfolio-name/trader mappings
portfolio_name_trader_mappings = [
# Benchmark trader
('Buy-and-hold Trader', Traders.BuyAndHoldTrader(), 'chartreuse'),
# Simple traders
('Simple Trader (perfect prediction)',
Traders.SimpleTrader_with_perfect_prediction(), 'pink'),
('Simple Trader (NN binary perfect prediction)',
Traders.SimpleTrader_with_nn_binary_perfect_prediction(), 'yellow'),
('Simple Trader (NN binary prediction)',
Traders.SimpleTrader_with_nn_binary_prediction(), 'brown'),
# Deep Q-Learning traders
('DQL Trader (perfect prediction)',
Traders.DqlTrader_with_perfect_prediction(), 'gray'),
('DQL Trader (NN binary perfect prediction)',
Traders.DqlTrader_with_nn_binary_perfect_prediction(), 'magenta'),
('DQL Trader (NN binary prediction)',