def export_linear_model_to_hdf(target_file, model_structure, estimator):
model_structure = _copy_model_structure(model_structure)
ols_model = memomodel.OLSModelDescription(intercept=estimator.intercept_,
coefs=estimator.coef_)
simmodel_description = memomodel.SimulationModelDescription(
model_structure=model_structure, regression_model=ols_model)
db = memomodel.MeMoSimDB(target_file)
db.open(access_mode=H5AccessMode.WRITE_TRUNCATE_ON_EXIST)
db.save_object(simmodel_description)
db.close()
def export_generic_sklearn_model_to_hdf(target_file, model_structure,
estimator):
model_structure = _copy_model_structure(model_structure)
generic_model = memomodel.GenericModelDescription(
sklearn_estimator=estimator)
simmodel_description = memomodel.SimulationModelDescription(
model_structure=model_structure, regression_model=generic_model)
db = memomodel.MeMoSimDB(target_file)
db.open(access_mode=H5AccessMode.WRITE_TRUNCATE_ON_EXIST)
db.save_object(simmodel_description)
db.close()
def export_kernel_ridge_regression_model_to_hdf(target_file, model_structure,
estimator):
model_structure = _copy_model_structure(model_structure)
krr_model = memomodel.KernelRidgeRegressionModelDescription(
kernel=estimator.kernel,
gamma=estimator.gamma,
degree=estimator.degree,
coef0=estimator.coef0,
X_fit=estimator.X_fit_,
dual_coef=estimator.dual_coef_)
simmodel_description = memomodel.SimulationModelDescription(
model_structure=model_structure, regression_model=krr_model)
db = memomodel.MeMoSimDB(target_file)
db.open(access_mode=H5AccessMode.WRITE_TRUNCATE_ON_EXIST)
db.save_object(simmodel_description)
db.close()