def test(data_path="dataset/", pars_choice=0, **kwargs): ### Local test log("#### Loading params ##############################################") model_pars, data_pars, compute_pars, out_pars = get_params(choice=pars_choice, data_path=data_path, **kwargs) print(model_pars, data_pars, compute_pars, out_pars) log("#### Loading dataset #############################################") dataset = get_dataset(data_pars) log("#### Model init, fit #############################################") from mlmodels.models import module_load_full, fit, predict module, model = module_load_full("model_keras.01_deepctr", model_pars, data_pars, compute_pars, dataset=dataset) model = fit(module, model, data_pars=data_pars, compute_pars=compute_pars, out_pars=out_pars, dataset=dataset) # log("#### Predict ####################################################") ypred = predict(module, model, compute_pars=compute_pars, data_pars=data_pars, out_pars=out_pars, dataset=dataset) log("#### metrics ####################################################") metrics_val = metrics(ypred, dataset[1], compute_pars=compute_pars, data_pars=data_pars, out_pars=out_pars) print(metrics_val) log("#### Plot #######################################################") log("#### Save/Load ##################################################") save_keras(model, save_pars=out_pars) from deepctr.layers import custom_objects model2 = load_keras(out_pars, custom_pars={"custom_objects": custom_objects}) model2.model.summary()
def save(model=None, save_pars=None, session=None): from mlmodels.util import save_keras print(save_pars) save_keras(model, session, save_pars=save_pars)
def save(model=None, session=None, save_pars={}): path = save_pars["outpath"] os.makedirs(path, exist_ok=True) save_keras(model, session, {"path": path + "/armdn.h5"})