add the nfl-transform-required columns into the train/val/test split

guidopetri · Mar 21, 2021 · 37c8cae · 37c8cae
1 parent b213597
commit 37c8cae
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/csgo_wp/data_transform.py b/csgo_wp/data_transform.py
@@ -365,6 +365,11 @@ def __init__(self,
                                       'AreaId',
                                       'IsAlive',
                                       'Side',
+                                      'Hp',
+                                      'Armor',
+                                      'EqValue',
+                                      'DistToBombsiteA',
+                                      'DistToBombsiteB',
                                       ])
 
             print('Getting match/map combinations...')