---
title: "R Notebook"
output: html_notebook
---

This notebook complements `train.R` by enabling the same functionality from the MLflow tutorial using an R notebook.

```{r}
library(mlflow)
library(glmnet)

set.seed(40)
```

Read the wine-quality csv file
 
```{r}
data <- read.csv("wine-quality.csv")
```

Split the data into training and test sets. (0.75, 0.25) split.

```{r}
sampled <- sample(1:nrow(data), 0.75 * nrow(data))
train <- data[sampled, ]
test <- data[-sampled, ]
```

The predicted column is "quality" which is a scalar from [3, 9]

```{r}
train_x <- as.matrix(train[, !(names(train) == "quality")])
test_x <- as.matrix(test[, !(names(train) == "quality")])
train_y <- train[, "quality"]
test_y <- test[, "quality"]
```

```{r}
alpha <- mlflow_param("alpha", 0.5, "numeric")
lambda <- mlflow_param("lambda", 0.5, "numeric")
```

```{r}
with(mlflow_start_run(), {
    model <- glmnet(train_x, train_y, alpha = alpha, lambda = lambda, family = "gaussian")
    predictor <- crate(~ glmnet::predict.glmnet(model, as.matrix(.x)), model)
    predicted <- predictor(test_x)

    rmse <- sqrt(mean((predicted - test_y) ^ 2))
    mae <- mean(abs(predicted - test_y))
    r2 <- cor(predicted, test_y) ^ 2

    message("Elasticnet model (alpha=", alpha, ", lambda=", lambda, "):")
    message("  RMSE: ", rmse)
    message("  MAE: ", mae)
    message("  R2: ", r2)

    mlflow_log_param("alpha", alpha)
    mlflow_log_param("lambda", lambda)
    mlflow_log_metric("rmse", rmse)
    mlflow_log_metric("r2", r2)
    mlflow_log_metric("mae", mae)

    mlflow_log_model(predictor, "model")
})
```

