Update train

kostya2011 · Nov 20, 2023 · 57781d0 · 57781d0
1 parent f2e8b55
commit 57781d0
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 12 deletions.
diff --git a/terraform/apps/app/app/variables.tf b/terraform/apps/app/app/variables.tf
@@ -34,7 +34,7 @@ variable "artifacts_bucket_name" {
 
 variable "model_name" {
   type    = string
-  default = "sk-learn-linear-regression-reg-model"
+  default = "sk-learn-random-forest-regressor-reg-model"
 }
 
 variable "model_version_number" {

diff --git a/train.py b/train.py
@@ -6,17 +6,6 @@
 import mlflow.sklearn
 import os
 
-#
-os.environ["MLFLOW_TRACKING_URI"] = "Change me"
-os.environ["MLFLOW_EXPERIMENT_NAME"] = "Change me"
-os.environ["MLFLOW_TRACKING_USERNAME"] = "Change me"
-os.environ["MLFLOW_TRACKING_PASSWORD"] = "Change me"
-
-# AWS AK/SK are required to upload artifacts to S3 Bucket
-os.environ["AWS_ACCESS_KEY_ID"] = "Change me"
-os.environ["AWS_SECRET_ACCESS_KEY"] = "Change me"
-
-
 def model_evaluate(model, test_features, test_labels):
     predictions = model.predict(test_features)
     error = mean_absolute_error(predictions, test_labels)