James McCool commited on
Commit
6c331fa
·
1 Parent(s): f4d2419

Update functions to support NBA data by replacing NFL references with NBA equivalents in regular and showdown functions.

Browse files
src/sim_func_hold/regular_functions.py CHANGED
@@ -9,23 +9,23 @@ def init_DK_seed_frames(slate_var, sharp_split):
9
 
10
 
11
  if slate_var == 'Main Slate':
12
- collection = db['DK_NFL_name_map']
13
  cursor = collection.find()
14
  raw_data = pd.DataFrame(list(cursor))
15
  names_dict = dict(zip(raw_data['key'], raw_data['value']))
16
- collection = db[f"DK_NFL_seed_frame_Main Slate"]
17
  elif slate_var == 'Secondary Slate':
18
- collection = db['DK_NFL_Secondary_name_map']
19
  cursor = collection.find()
20
  raw_data = pd.DataFrame(list(cursor))
21
  names_dict = dict(zip(raw_data['key'], raw_data['value']))
22
- collection = db[f"DK_NFL_seed_frame_Secondary Slate"]
23
  elif slate_var == 'Late Slate':
24
- collection = db['DK_NFL_Late_name_map']
25
  cursor = collection.find()
26
  raw_data = pd.DataFrame(list(cursor))
27
  names_dict = dict(zip(raw_data['key'], raw_data['value']))
28
- collection = db[f"DK_NFL_seed_frame_Late Slate"]
29
 
30
  cursor = collection.find().limit(sharp_split)
31
 
 
9
 
10
 
11
  if slate_var == 'Main Slate':
12
+ collection = db['DK_NBA_name_map']
13
  cursor = collection.find()
14
  raw_data = pd.DataFrame(list(cursor))
15
  names_dict = dict(zip(raw_data['key'], raw_data['value']))
16
+ collection = db[f"DK_NBA_seed_frame_Main Slate"]
17
  elif slate_var == 'Secondary Slate':
18
+ collection = db['DK_NBA_Secondary_name_map']
19
  cursor = collection.find()
20
  raw_data = pd.DataFrame(list(cursor))
21
  names_dict = dict(zip(raw_data['key'], raw_data['value']))
22
+ collection = db[f"DK_NBA_seed_frame_Secondary Slate"]
23
  elif slate_var == 'Late Slate':
24
+ collection = db['DK_NBA_Late_name_map']
25
  cursor = collection.find()
26
  raw_data = pd.DataFrame(list(cursor))
27
  names_dict = dict(zip(raw_data['key'], raw_data['value']))
28
+ collection = db[f"DK_NBA_seed_frame_Late Slate"]
29
 
30
  cursor = collection.find().limit(sharp_split)
31
 
src/sim_func_hold/showdown_functions.py CHANGED
@@ -35,7 +35,7 @@ def init_SD_baselines(slate_var):
35
 
36
  raw_display = pd.DataFrame(list(cursor))
37
  raw_display = raw_display[raw_display['version'] == 'overall']
38
- raw_display = raw_display[['Player', 'Position', 'Team', 'Opp', 'Salary', 'Floor', 'Median', 'Ceiling', 'Top_finish', 'Top_5_finish', 'Top_10_finish', '20+%', '2x%', '3x%', '4x%',
39
  'Own', 'Small_Field_Own', 'Large_Field_Own', 'Cash_Field_Own', 'CPT_Own', 'LevX', 'version', 'slate', 'timestamp', 'player_id', 'site']]
40
  raw_display = raw_display.rename(columns={'player_id': 'player_ID'})
41
  raw_display['Small_Field_Own'] = raw_display['Large_Field_Own']
 
35
 
36
  raw_display = pd.DataFrame(list(cursor))
37
  raw_display = raw_display[raw_display['version'] == 'overall']
38
+ raw_display = raw_display[['Player', 'Position', 'Team', 'Opp', 'Salary', 'Floor', 'Median', 'Ceiling', 'Top_finish', 'Top_5_finish', 'Top_10_finish', '20+%', '4x%', '5x%', '6x%', 'GPP%',
39
  'Own', 'Small_Field_Own', 'Large_Field_Own', 'Cash_Field_Own', 'CPT_Own', 'LevX', 'version', 'slate', 'timestamp', 'player_id', 'site']]
40
  raw_display = raw_display.rename(columns={'player_id': 'player_ID'})
41
  raw_display['Small_Field_Own'] = raw_display['Large_Field_Own']