Update app.R

app.R

@@ -4,12 +4,18 @@ library(data.table)
 library(xgboost)
 library(caret)
 library(shinythemes)
+library(ggplot2)
+
 calculate_timstuff <- function(data, pitch_type) {
     data_matrix <- as.matrix(data)
     if (pitch_type == "Offspeed") {
       mod <- xgb.load('Off.model')
       prediction <- predict(mod, data_matrix)
       timstuff <- (50 - (prediction - -0.002239657) / 0.01043216 * 10)
+    } else if (pitch_type == "Breaking") {
+      mod <- xgb.load('Break.model')
+      prediction <- predict(mod, data_matrix)
+      timstuff <- (50 - (prediction - -0.004822031) / 0.007912765 * 10)
     } else {
       mod <- xgb.load('FB.model')
       prediction <- predict(mod, data_matrix)
@@ -17,17 +23,16 @@ calculate_timstuff <- function(data, pitch_type) {
     }
     return(timstuff)
   }
+
 ui <- fluidPage(
   theme = shinytheme("flatly"),
   
-  # Application title
-  titlePanel("TimStuff Fastball/Offspeed Calculator"),
+  titlePanel("TimStuff Fastball/Offspeed/Breaking Calculator"),
   
-  # Sidebar with inputs
   sidebarLayout(
     sidebarPanel(
       width = 4,
-      radioButtons("pitch", "Pitch Type", choices = c("Fastball", "Offspeed"), inline = TRUE),
+      radioButtons("pitch", "Pitch Type", choices = c("Fastball", "Offspeed", "Breaking"), inline = TRUE),
       radioButtons("phand", "Hand", choices = c("R", "L"), inline = TRUE),
       
       numericInput("velo", "Velocity (mph)", value = 90, min = 70, max = 110, step = 1),
@@ -44,18 +49,15 @@ ui <- fluidPage(
                         "Horizontal Break" = "HB",
                         "Spin Rate" = "spin_rate",
                         "Extension" = "EAA",
-                        "Spin Axis Difference" = "SADiff",
-                        "Horizontal Release Point" = "x0",
-                        "Vertical Release Point" = "z0"))
+                        "Spin Axis Difference" = "SADiff"))
     ),
     
-    # Main panel with output
     mainPanel(
       width = 8,
       wellPanel(
         h3("Results"),
         verbatimTextOutput("stuff"),
-        plotOutput("feature_impact", height = "400px")  # Placeholder for future visualization
+        plotOutput("feature_impact", height = "400px")
       )
     )
   )
@@ -67,7 +69,7 @@ server <- function(input, output) {
       data <- data.frame(
         start_speed = input$velo,
         IVB = input$ivb,
-        HB = ifelse(input$phand == "L", -input$hb, input$hb),
+        HB = input$hb,
         EAA = input$ext / 6.3,
         x0 = ifelse(input$phand == "L", -input$x0, input$x0),
         z0 = input$z0,
@@ -77,79 +79,59 @@ server <- function(input, output) {
       
       data <- as.matrix(data)
       
-      if (input$pitch == "Offspeed") {
-        mod <- xgb.load('Off.model')
-        prediction <- predict(mod, data)
-        TimStuff <- (50 - (prediction - -0.002239657) / 0.01043216 * 10)
-      } else {
-        mod <- xgb.load('FB.model')
-        prediction <- predict(mod, data)
-        TimStuff <- (50 - (prediction - -0.0011801) / 0.007989927 * 10)
-      }
+      TimStuff <- calculate_timstuff(data, input$pitch)
       
       paste("TimStuff:", round(TimStuff, 2), "\n20-80 Scale, 50 is Average, 1 SD is 10")
     })
     
-    # Add this new reactive expression
-    # Add this new reactive expression
-  feature_range <- reactive({
-    ranges <- list(
-      start_speed = c(70, 110),
-      IVB = c(-20, 30),
-      HB = c(-20, 20),
-      spin_rate = c(1000, 3500),
-      EAA = c(5/6.3, 8/6.3),
-      SADiff = c(-180, 180),
-      x0 = c(-5, 5),
-      z0 = c(3, 8)
-    )
-    ranges[[input$feature_to_vary]]
-  })
+    feature_range <- reactive({
+      ranges <- list(
+        start_speed = c(70, 110),
+        IVB = c(-20, 30),
+        HB = c(-20, 20),
+        spin_rate = c(1000, 3500),
+        EAA = c(5/6.3, 8/6.3),
+        SADiff = c(-180, 180)
+      )
+      ranges[[input$feature_to_vary]]
+    })
 
-  # Update the feature_impact plot
-  output$feature_impact <- renderPlot({
-    # Create base data frame with current inputs
-    base_data <- data.frame(
-      start_speed = input$velo,
-      IVB = input$ivb,
-      HB = ifelse(input$phand == "L", -input$hb, input$hb),
-      EAA = input$ext / 6.3,
-      x0 = ifelse(input$phand == "L", -input$x0, input$x0),
-      z0 = input$z0,
-      spin_rate = input$spinrate,
-      SADiff = input$spinaxisdiff
-    )
-    
-    # Generate sequence for the selected feature
-    feature_seq <- seq(feature_range()[1], feature_range()[2], length.out = 100)
-    
-    # Calculate TimStuff for the sequence
-    varied_data <- do.call(rbind, replicate(100, base_data, simplify = FALSE))
-    varied_data[[input$feature_to_vary]] <- feature_seq
-    timstuff <- calculate_timstuff(varied_data, input$pitch)
-    
-    # Create plot data
-    plot_data <- data.frame(
-      Feature = names(base_data)[which(names(base_data) == input$feature_to_vary)],
-      Value = feature_seq,
-      TimStuff = timstuff
-    )
-    
-    # Create plot
-    ggplot(plot_data, aes(x = Value, y = TimStuff)) +
-      geom_line(color = "blue", size = 1) +
-      geom_point(aes(x = base_data[[input$feature_to_vary]], 
-                     y = calculate_timstuff(base_data, input$pitch)),
-                 color = "red", size = 3) +
-      theme_minimal() +
-      ylim(0,100) +
-      labs(title = paste("Impact of", input$feature_to_vary, "on TimStuff"),
-           x = input$feature_to_vary, y = "TimStuff") +
-      theme(text = element_text(size = 14))
-  })
+    output$feature_impact <- renderPlot({
+      base_data <- data.frame(
+        start_speed = input$velo,
+        IVB = input$ivb,
+        HB = input$hb,
+        EAA = input$ext / 6.3,
+        x0 = ifelse(input$phand == "L", -input$x0, input$x0),
+        z0 = input$z0,
+        spin_rate = input$spinrate,
+        SADiff = input$spinaxisdiff
+      )
+      
+      feature_seq <- seq(feature_range()[1], feature_range()[2], length.out = 100)
+      
+      varied_data <- do.call(rbind, replicate(100, base_data, simplify = FALSE))
+      varied_data[[input$feature_to_vary]] <- feature_seq
+      timstuff <- calculate_timstuff(varied_data, input$pitch)
+      
+      plot_data <- data.frame(
+        Feature = names(base_data)[which(names(base_data) == input$feature_to_vary)],
+        Value = feature_seq,
+        TimStuff = timstuff
+      )
+      
+      ggplot(plot_data, aes(x = Value, y = TimStuff)) +
+        geom_line(color = "blue", size = 1) +
+        geom_point(aes(x = base_data[[input$feature_to_vary]], 
+                       y = calculate_timstuff(base_data, input$pitch)),
+                   color = "red", size = 3) +
+        theme_minimal() +
+        ylim(0,100) +
+        labs(title = paste("Impact of", input$feature_to_vary, "on TimStuff"),
+             x = input$feature_to_vary, y = "TimStuff") +
+        theme(text = element_text(size = 14))
+    })
   })
 }
 
-# Run the application 
-shinyApp(ui = ui, server = server)
-
+shinyApp(ui = ui, server = server)