import clean_data as cd
import clean_data_n as cdn
import gen_submission as gs
import pandas as pd


test_l = ["N", "O", "P", "Q", "R", "S"]
seed_data = pd.DataFrame.from_csv("/media/3rd/git_data/kaggle_contests/" +
                                  "march_madness/tourney_seeds.csv",
                                  index_col=False)
cmet_data = pd.DataFrame.from_csv("/media/3rd/git_data/kaggle_contests/" +
                                  "march_madness/chessmetrics.csv",
                                  index_col=False)
teams = cdn.genSeasonTeams(seed_data)
cmet_data = cd.cleanCMet(cmet_data)
prob_mat = gs.genProbMat(test_l, teams, cmet_data)
gs.writeResults(prob_mat, "/media/3rd/git_data/kaggle_contests" +
                "/march_madness/test_submssion.csv")
