-
Notifications
You must be signed in to change notification settings - Fork 28
/
train_model.py
30 lines (23 loc) · 1.04 KB
/
train_model.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
#!/usr/bin/env python
# Do *not* edit this script. Changes will be discarded so that we can train the models consistently.
# This file contains functions for training models for the Challenge. You can run it as follows:
#
# python train_model.py data model
#
# where 'data' is a folder containing the Challenge data and 'model' is a folder for saving your model.
import sys
from helper_code import is_integer
from team_code import train_challenge_model
if __name__ == '__main__':
# Parse the arguments.
if not (len(sys.argv) == 3 or len(sys.argv) == 4):
raise Exception('Include the data and model folders as arguments, e.g., python train_model.py data model.')
# Define the data and model foldes.
data_folder = sys.argv[1]
model_folder = sys.argv[2]
# Change the level of verbosity; helpful for debugging.
if len(sys.argv)==4 and is_integer(sys.argv[3]):
verbose = int(sys.argv[3])
else:
verbose = 1
train_challenge_model(data_folder, model_folder, verbose) ### Teams: Implement this function!!!