args = parser.parse_args()
# ====== Open data file ======
file_name = ''
if args.year == 2017:
file_name = DATA_17
elif args.year == 2016:
file_name = DATA_16
f = open(DIRNAME + "/" + file_name, 'r')
print("Loading JSON data")
data = json.load(f)
print("Preprocessing the data")
# formatter class that contains trainX, trainY, testX, testY for individual teams
formatter = Formatter(data, args.train_size, args.sequence_length)
trainX_home, trainX_away, trainY, testX_home, testX_away, testY = formatter.get_data(
)
## ======== Build model ======
with tf.Session() as sess:
kbo_pred_model = SeLuModel(sess,
args.model_name,
learn_rate=args.learn_rate,
sequence_length=args.sequence_length)
## ======== Train model ======
print("Started the training...")
kbo_runner = Runner()
kbo_runner.train_run(kbo_pred_model,
trainX_home,
trainX_away,
trainY,
