Added stack argument to LineupGenerator class and different class options to Nba subclass. Added input for feb 3

Author: ghazi-arman

LineupGenerator.py

@@ -7,12 +7,13 @@
 import os.path
 
 class LineupGenerator:
-	def __init__(self, sport, num_lineups, overlap, player_limit, teams_limit, solver, correlation_file, players_file, defenses_goalies_file, output_file):
+	def __init__(self, sport, num_lineups, overlap, player_limit, teams_limit, stack, solver, correlation_file, players_file, defenses_goalies_file, output_file):
 		self.sport = sport
 		self.num_lineups = num_lineups
 		self.overlap = overlap
 		self.player_limit = player_limit
 		self.teams_limit = teams_limit
+		self.stack = stack
 		self.solver = solver
 		self.players = self.load(players_file)
 		self.num_players = len(self.players.index)

Nba.py

@@ -2,8 +2,8 @@
 from LineupGenerator import LineupGenerator
 
 class Nba(LineupGenerator):
-	def __init__(self, sport, num_lineups, overlap, player_limit, teams_limit, solver, correlation_file, players_file, defenses_goalies_file, output_file):
-		super().__init__(sport, num_lineups, overlap, player_limit, teams_limit, solver, correlation_file, players_file, defenses_goalies_file, output_file)
+	def __init__(self, sport, num_lineups, overlap, player_limit, teams_limit, stack, solver, correlation_file, players_file, defenses_goalies_file, output_file):
+		super().__init__(sport, num_lineups, overlap, player_limit, teams_limit, stack, solver, correlation_file, players_file, defenses_goalies_file, output_file)
 		self.salary_cap = 55000
 		self.header = ['PG', 'PG', 'SG', 'SG', 'SF', 'SF', 'PF', 'PF', 'C']
 
@@ -31,10 +31,51 @@ def generate(self, lineups):
 			prob += (used_team[i] <= (pulp.lpSum(self.players_teams[k][i]*players_lineup[k] for k in range(self.num_players))))
 
 			# stacks SGs with another player from the same team
-			prob += (pulp.lpSum(self.players_teams[k][i]*self.positions['PG'][k]*players_lineup[k] for k in range(self.num_players))
-				<= pulp.lpSum(self.players_teams[k][i]*self.positions['SG'][k]*players_lineup[k] for k in range(self.num_players)))
+			if self.stack == "PG-SG":
+				prob += (pulp.lpSum(self.players_teams[k][i]*self.positions['PG'][k]*players_lineup[k] for k in range(self.num_players))
+					<= pulp.lpSum(self.players_teams[k][i]*self.positions['SG'][k]*players_lineup[k] for k in range(self.num_players)))
+			elif self.stack == "PG-(SG-SF-PF)":
+				prob += (pulp.lpSum(self.players_teams[k][i]*self.positions['PG'][k]*players_lineup[k] for k in range(self.num_players))
+					<= (pulp.lpSum(self.players_teams[k][i]*self.positions['SG'][k]*players_lineup[k] for k in range(self.num_players))
+					+ pulp.lpSum(self.players_teams[k][i]*self.positions['SF'][k]*players_lineup[k] for k in range(self.num_players))
+					+ pulp.lpSum(self.players_teams[k][i]*self.positions['PF'][k]*players_lineup[k] for k in range(self.num_players))))
+			elif self.stack == "PG-(SG-SF-PF)-(SG-SF-PF)":
+				prob += (2*pulp.lpSum(self.players_teams[k][i]*self.positions['PG'][k]*players_lineup[k] for k in range(self.num_players))
+					<= (pulp.lpSum(self.players_teams[k][i]*self.positions['SG'][k]*players_lineup[k] for k in range(self.num_players))
+					+ pulp.lpSum(self.players_teams[k][i]*self.positions['SF'][k]*players_lineup[k] for k in range(self.num_players))
+					+ pulp.lpSum(self.players_teams[k][i]*self.positions['PF'][k]*players_lineup[k] for k in range(self.num_players))))
+			elif self.stack == "PG-SG+PG-SG":
+				for j in range(self.num_teams):
+					if i != j:
+						prob += (pulp.lpSum(self.players_teams[k][i]*self.positions['PG'][k]*players_lineup[k] for k in range(self.num_players))
+							<= pulp.lpSum(self.players_teams[k][i]*self.positions['SG'][k]*players_lineup[k] for k in range(self.num_players)))
+						prob += (pulp.lpSum(self.players_teams[k][j]*self.positions['PG'][k]*players_lineup[k] for k in range(self.num_players))
+							<= pulp.lpSum(self.players_teams[k][j]*self.positions['SG'][k]*players_lineup[k] for k in range(self.num_players)))
+			elif self.stack == "PG-(SG-SF-PF)+PG-(SG-SF-PF)":
+				for j in range(self.num_teams):
+					if i != j:
+						prob += (pulp.lpSum(self.players_teams[k][i]*self.positions['PG'][k]*players_lineup[k] for k in range(self.num_players))
+							<= (pulp.lpSum(self.players_teams[k][i]*self.positions['SG'][k]*players_lineup[k] for k in range(self.num_players))
+							+ pulp.lpSum(self.players_teams[k][i]*self.positions['SF'][k]*players_lineup[k] for k in range(self.num_players))
+							+ pulp.lpSum(self.players_teams[k][i]*self.positions['PF'][k]*players_lineup[k] for k in range(self.num_players))))
+						prob += (pulp.lpSum(self.players_teams[k][j]*self.positions['PG'][k]*players_lineup[k] for k in range(self.num_players))
+							<= (pulp.lpSum(self.players_teams[k][j]*self.positions['SG'][k]*players_lineup[k] for k in range(self.num_players))
+							+ pulp.lpSum(self.players_teams[k][j]*self.positions['SF'][k]*players_lineup[k] for k in range(self.num_players))
+							+ pulp.lpSum(self.players_teams[k][j]*self.positions['PF'][k]*players_lineup[k] for k in range(self.num_players))))
+			elif self.stack == "PG-(SG-SF-PF)-(SG-SF-PF)+PG-(SG-SF-PF)-(SG-SF-PF)":
+				for j in range(self.num_teams):
+					if i != j:
+						prob += (2*pulp.lpSum(self.players_teams[k][i]*self.positions['PG'][k]*players_lineup[k] for k in range(self.num_players))
+							<= (pulp.lpSum(self.players_teams[k][i]*self.positions['SG'][k]*players_lineup[k] for k in range(self.num_players))
+							+ pulp.lpSum(self.players_teams[k][i]*self.positions['SF'][k]*players_lineup[k] for k in range(self.num_players))
+							+ pulp.lpSum(self.players_teams[k][i]*self.positions['PF'][k]*players_lineup[k] for k in range(self.num_players))))
+						prob += (2*pulp.lpSum(self.players_teams[k][j]*self.positions['PG'][k]*players_lineup[k] for k in range(self.num_players))
+							<= (pulp.lpSum(self.players_teams[k][j]*self.positions['SG'][k]*players_lineup[k] for k in range(self.num_players))
+							+ pulp.lpSum(self.players_teams[k][j]*self.positions['SF'][k]*players_lineup[k] for k in range(self.num_players))
+							+ pulp.lpSum(self.players_teams[k][j]*self.positions['PF'][k]*players_lineup[k] for k in range(self.num_players))))
+			
 			prob += (pulp.lpSum(self.players_teams[k][i]*players_lineup[k] for k in range(self.num_players)) <= 4*used_team[i])
-		# # ensures that the lineup contains less than X unique teams
+		# ensures that the lineup contains less than X unique teams
 		prob += (pulp.lpSum(used_team[i] for i in range(self.num_teams)) == 5)
 
 		# each new lineup can't have more than the overlap variable number of combinations of players in any previous lineups

main.py

@@ -13,6 +13,7 @@
 			overlap = int(sys.argv[4]),
 			player_limit = int(sys.argv[5]),
 			teams_limit = int(sys.argv[6]),
+			stack = sys.argv[7],
 			solver = pulp.GLPK_CMD(msg=0),
 			correlation_file = 'nfl/inputs/{}/correlation.csv'.format(sys.argv[2]),
 			players_file = 'nfl/inputs/players.csv',
@@ -26,6 +27,7 @@
 			overlap = int(sys.argv[4]),
 			player_limit = int(sys.argv[5]),
 			teams_limit = int(sys.argv[6]),
+			stack = sys.argv[7],
 			solver = pulp.GLPK_CMD(msg=0),
 			correlation_file = 'nhl/inputs/{}/correlation.csv'.format(sys.argv[2]),
 			players_file = 'nhl/inputs/players.csv',
@@ -39,6 +41,7 @@
 			overlap = int(sys.argv[4]),
 			player_limit = int(sys.argv[5]),
 			teams_limit = int(sys.argv[6]),
+			stack = sys.argv[7],
 			solver = pulp.GLPK_CMD(msg=0),
 			correlation_file = 'nba/inputs/{}/correlation.csv'.format(sys.argv[2]),
 			players_file = 'nba/inputs/players.csv',

multiple.sh

@@ -1,14 +1,20 @@
 #!/bin/sh
-overlaps=(3 4 5)
-player_limits=(10 30 75)
+overlaps=(4 5 6)
+player_limits=(50 150)
 team_limits=(3 4 5 6)
+stacks=("PG-SG" "PG-(SG-SF-PF)" "PG-(SG-SF-PF)-(SG-SF-PF)" "PG-SG+PG-SG" "PG-(SG-SF-PF)+PG-(SG-SF-PF)" "PG-(SG-SF-PF)-(SG-SF-PF)+PG-(SG-SF-PF)-(SG-SF-PF)")
 for overlap in "${overlaps[@]}"
 do
   for player_limit in "${player_limits[@]}"
   do
     for team_limit in "${team_limits[@]}"
     do
-      ./run.sh $1 $2 $3 $overlap $player_limit $team_limit $4
+      for stack in "${stacks[@]}"
+      do
+        if [ ! -e "$1"/outputs/"$2"/output_"$3"_overlap_"$overlap"_playerlimit_"$player_limit"_numteams_"$team_limit"_stack_"$stack"_proj.csv ]; then
+          ./run.sh $1 $2 $3 $overlap $player_limit $team_limit $stack
+        fi
+      done
     done
   done
 done