knitr::opts_chunk$set(echo = TRUE)
#install.packages("caret")
library(caret)
library(EHData)
data <- read.csv("D:\\Rstudio\\Cuny_622\\Space\\train.csv")
test_data <- read.csv("D:\\Rstudio\\Cuny_622\\Space\\test.csv")
set.seed(123)

model <- EHModel_Regression_Logistic(data, "Transported")

predictions <- predict(model, test_data)
confusion_matrix <- confusionMatrix(predictions, test_data$Transported)
print(confusion_matrix)


ericonsi/EHData documentation built on April 12, 2025, 7:39 p.m.