team-data-import-ml4k-numbers

gistfile1.txt

import csv
import ml4k
import os

listOfFiles = [
    './nflscrapR-data/games_data/regular_season/reg_games_2018.csv',
    './nflscrapR-data/games_data/regular_season/reg_games_2017.csv',
    './nflscrapR-data/games_data/regular_season/reg_games_2016.csv',
    './nflscrapR-data/games_data/regular_season/reg_games_2015.csv'
]

API_KEY = "****************************************"
model = ml4k.Model(API_KEY)

def get_nfs_team_number(team_name):
  if team_name == 'SD':
      team_name = 'LAC'
  if team_name == 'JAC':
      team_name = 'JAX'
  if team_name == 'STL':
      team_name = 'LA'

  nfl_teams = ['ARI','ATL','BAL','BUF','CAR', 'CHI','CIN','CLE','DAL','DEN','DET','GB','HOU','IND','JAX','KC','LA','LAC','MIA','MIN','NE','NO','NYG','NYJ','OAK','PHI','PIT','SEA','SF','TB','TEN','WAS']
  return nfl_teams.index(team_name)


for file in listOfFiles: 
    with open(file) as csv_file:
        csv_reader = csv.reader(csv_file, delimiter=',')
        line_count = 0
        for row in csv_reader:
        
            if line_count != 0:
                home_team = get_nfs_team_number(row[2])
                away_team = get_nfs_team_number(row[3])
                home_score = row[8]
                away_score = row[9]

                if home_score > away_score:
                    winner = 'home_team'
                if home_score < away_score:
                    winner = 'away_team'

                matchup = [ home_team, away_team ]    
                print( 'home_team: ' + str(home_team))
                print( 'away_team: ' + str(away_team))
                print( 'label: ' + winner)
                    
                    # model.add_training_data(label, board_state)
                model.add_training_data(winner, matchup)

            line_count += 1
                    
            print(f'Processed {line_count} lines.')