Una vez que tengas tus conjuntos de entrenamiento y prueba aleatorios, puedes ajustar un modelo de regresión logística al conjunto de entrenamiento usando la función glm(). glm() es una versión más avanzada de lm() que permite modelos de regresión más variados, más allá de la regresión lineal por mínimos cuadrados ordinarios.

Acuérdate de pasar el argumento family = "binomial" a glm() para indicar que quieres hacer regresión logística (en lugar de lineal). Por ejemplo:

glm(Target ~ ., family = "binomial", dataset)

No te preocupes por advertencias como glm.fit: algorithm did not converge o glm.fit: fitted probabilities numerically 0 or 1 occurred. Son habituales en conjuntos de datos pequeños y, por lo general, no causan problemas. Normalmente significan que tu conjunto de datos es perfectamente separable, lo que puede generar dificultades para las matemáticas del modelo, pero la función glm() de R casi siempre es lo bastante robusta como para manejar este caso sin problemas.

Cuando tengas un modelo glm() ajustado a tu conjunto de datos, puedes predecir el resultado (p. ej., roca o mina) sobre el conjunto test usando la función predict() con el argumento type = "response":

predict(my_model, test, type = "response")

¿Por qué dividir en train/test?

Machine Learning con caret en R

Ejercicio interactivo práctico