forked from tradytics/eiten
-
Notifications
You must be signed in to change notification settings - Fork 0
/
strategy_manager.py
60 lines (54 loc) · 2.65 KB
/
strategy_manager.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
# Basic libraries
import os
import warnings
from strategies.genetic_algo_strategy import GeneticAlgoStrategy
from strategies.maximum_sharpe_ratio_strategy import MaximumSharpeRatioStrategy
from strategies.eigen_portfolio_strategy import EigenPortfolioStrategy
from strategies.minimum_variance_strategy import MinimumVarianceStrategy
from strategies.strategy_helper_functions import StrategyHelperFunctions
warnings.filterwarnings("ignore")
class StrategyManager:
"""
Runs and manages all strategies
"""
def __init__(self):
print("\n--= Strategy manager has been created...")
self.geneticAlgoStrategy = GeneticAlgoStrategy()
self.minimumVarianceStrategy = MinimumVarianceStrategy()
self.eigenPortfolioStrategy = EigenPortfolioStrategy()
self.maximumSharpeRatioStrategy = MaximumSharpeRatioStrategy()
self.strategyHelperFunctions = StrategyHelperFunctions()
def calculate_genetic_algo_portfolio(self, symbols, returns_matrix_percentages):
"""
Genetic algorithm based portfolio that maximizes sharpe ratio. This is my own implementation
"""
print("-* Calculating portfolio weights using genetic algorithm...")
portfolio_weights_dictionary = self.geneticAlgoStrategy.generate_portfolio(symbols, returns_matrix_percentages)
return portfolio_weights_dictionary
def calculate_eigen_portfolio(self, symbols, covariance_matrix, eigen_portfolio_number):
"""
2nd Eigen Portfolio
"""
print("-$ Calculating portfolio weights using eigen values...")
portfolio_weights_dictionary = self.eigenPortfolioStrategy.generate_portfolio(symbols, covariance_matrix, eigen_portfolio_number)
return portfolio_weights_dictionary
def calculate_minimum_variance_portfolio(self, symbols, covariance_matrix):
"""
Minimum variance portfolio
"""
print("-! Calculating portfolio weights using minimum variance portfolio algorithm...")
portfolio_weights_dictionary = self.minimumVarianceStrategy.generate_portfolio(symbols, covariance_matrix)
return portfolio_weights_dictionary
def calculate_maximum_sharpe_portfolio(self, symbols, covariance_matrix, returns_vector):
"""
Maximum sharpe portfolio
"""
print("-# Calculating portfolio weights using maximum sharpe portfolio algorithm...")
portfolio_weights_dictionary = self.maximumSharpeRatioStrategy.generate_portfolio(symbols, covariance_matrix, returns_vector)
return portfolio_weights_dictionary
def random_matrix_theory_based_cov(self, returns_matrix):
"""
Covariance matrix filtering using random matrix theory
"""
filtered_covariance_matrix = self.strategyHelperFunctions.random_matrix_theory_based_cov(returns_matrix)
return filtered_covariance_matrix