parser.add_argument('--cf')
args = parser.parse_args()
collection_name = args.cn
model_name = args.mn
model_type = args.mt
config_path = args.cf
config = ConfigParser.ConfigParser()
config.read(config_path)
root_logger.info('Get train data.')
data_rep = DataRepository(uri=config.get('defaults', 'MONGO_URI'))
X = []
Y = []
for row in data_rep.get(collection_name, { '$or': [{'Y': [1]}, {'Y': [0]}]}):
root_logger.info(row['meta'])
X.append(row['X'])
Y.append(row['Y'])
root_logger.info('We have {0} rows to train'.format(len(Y)))
root_logger.info('Train model.')
model = StatModelFactory.create(model_type)
model.train(X, Y)
model_storage = AzureBlobStorage(
config.get('defaults', 'AZURE_STORAGE_NAME'),
config.get('defaults', 'AZURE_STORAGE_KEY'),
'deepsportmodels')
