def make_initialize_job(model_json, raw_filepath=None, uuid=None):
initialize_job = DMLJob(JobTypes.JOB_INIT.name,
model_json,
"keras",
hyperparams=make_hyperparams(),
label_column_name='label',
raw_filepath=raw_filepath,
uuid=uuid)
return initialize_job
def make_initialize_job(model_json):
"""
Helper function to make an DMLJob of 'initialize' type.
"""
initialize_job = DMLJob(
"initialize",
model_json,
"keras"
)
return initialize_job
def make_validate_job(model_json, new_weights, hyperparams, \
session_filepath, datapoint_count):
validate_job = DMLJob(JobTypes.JOB_VAL.name,
model_json,
'keras',
new_weights,
hyperparams,
'label',
session_filepath=session_filepath,
datapoint_count=datapoint_count)
return validate_job
def make_train_job(model_json, initial_weights, hyperparams, \
session_filepath, datapoint_count):
train_job = DMLJob(JobTypes.JOB_TRAIN.name,
model_json,
"keras",
initial_weights,
hyperparams,
'label',
session_filepath=session_filepath,
datapoint_count=datapoint_count)
return train_job
def make_split_job( \
model_json, raw_filepath):
split_job = DMLJob(
JobTypes.JOB_SPLIT.name,
model_json,
"keras",
hyperparams=make_hyperparams(),
label_column_name='label',
raw_filepath=raw_filepath,
)
return split_job
def make_communicate_job(key, weights):
communicate_job = DMLJob(JobTypes.JOB_COMM.name, key, "keras")
communicate_job.key = key
return communicate_job