Change experiment name
This commit is contained in:
parent
6875182306
commit
0a26997cd5
4
train.py
4
train.py
@ -21,7 +21,7 @@ logging.basicConfig(level=logging.WARN)
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
mlflow.set_tracking_uri("http://172.17.0.1:5000")
|
||||
mlflow.set_experiment("s123456")
|
||||
mlflow.set_experiment("s123456-new")
|
||||
|
||||
def eval_metrics(actual, pred):
|
||||
rmse = np.sqrt(mean_squared_error(actual, pred))
|
||||
@ -94,6 +94,6 @@ if __name__ == "__main__":
|
||||
# There are other ways to use the Model Registry, which depends on the use case,
|
||||
# please refer to the doc for more information:
|
||||
# https://mlflow.org/docs/latest/model-registry.html#api-workflow
|
||||
mlflow.sklearn.log_model(lr, "wines-model", registered_model_name="ElasticnetWineModel", signature=signature)
|
||||
mlflow.sklearn.log_model(lr, "wines-model", registered_model_name="ElasticnetWineModel_s123456", signature=signature)
|
||||
else:
|
||||
mlflow.sklearn.log_model(lr, "model", signature=signature)
|
||||
|
Loading…
Reference in New Issue
Block a user