Hola Agustina,
En líneas generales si, pero dejame aclarar un par de cosas. En el fit es tal cual como vos decís, la idea es calcular w y b a partir de los datos (asociado a la etapa de entrenamiento).
El predict está asociado a la etapa de testeo, pero básicamente predice que Y (duración de su batería) le corresponde a cada X que le pases (capacidad del smartphone). Si bien lo típico es usarlo en testeo, nada quita que se pueda usar con el conjunto de datos de entrenamiento (por ejemplo para calcular el riesgo empírico). Lo importante es que lo estás haciendo una predicción.
En cuanto a como graficar la recta, eso deberías hacerlo aparte (fuera de class). ¿Como? Bueno, una recta se determina con 2 puntos; basta con hacer un predict a dos X cualesquiera y unirlos con una recta.
Éxitos!
En líneas generales si, pero dejame aclarar un par de cosas. En el fit es tal cual como vos decís, la idea es calcular w y b a partir de los datos (asociado a la etapa de entrenamiento).
El predict está asociado a la etapa de testeo, pero básicamente predice que Y (duración de su batería) le corresponde a cada X que le pases (capacidad del smartphone). Si bien lo típico es usarlo en testeo, nada quita que se pueda usar con el conjunto de datos de entrenamiento (por ejemplo para calcular el riesgo empírico). Lo importante es que lo estás haciendo una predicción.
En cuanto a como graficar la recta, eso deberías hacerlo aparte (fuera de class). ¿Como? Bueno, una recta se determina con 2 puntos; basta con hacer un predict a dos X cualesquiera y unirlos con una recta.
Éxitos!