Afficher y = mx + b
Dans cette étape, vous allez apprendre à afficher l'équation de régression linéaire et à prédire les valeurs de y en utilisant la pente et l'ordonnée à l'origine calculées.
Mettez à jour le fichier linear_regression.c pour ajouter la fonctionnalité de prédiction :
cd ~/project
nano linear_regression.c
Remplacez le code précédent par l'implémentation suivante :
#include <stdio.h>
#include <math.h>
#define MAX_POINTS 100
typedef struct {
double x;
double y;
} DataPoint;
void computeLinearRegression(DataPoint points[], int num_points, double *m, double *b) {
double sum_x = 0, sum_y = 0, sum_xy = 0, sum_x_squared = 0;
for (int i = 0; i < num_points; i++) {
sum_x += points[i].x;
sum_y += points[i].y;
sum_xy += points[i].x * points[i].y;
sum_x_squared += points[i].x * points[i].x;
}
double n = num_points;
*m = (n * sum_xy - sum_x * sum_y) / (n * sum_x_squared - sum_x * sum_x);
*b = (sum_y - (*m) * sum_x) / n;
}
// Function to predict y value
double predictY(double m, double b, double x) {
return m * x + b;
}
int main() {
DataPoint points[MAX_POINTS];
int num_points = 0;
printf("Enter x and y coordinates (enter -1 -1 to finish):\n");
while (num_points < MAX_POINTS) {
double x, y;
scanf("%lf %lf", &x, &y);
if (x == -1 && y == -1) {
break;
}
points[num_points].x = x;
points[num_points].y = y;
num_points++;
}
double slope, intercept;
computeLinearRegression(points, num_points, &slope, &intercept);
printf("\nLinear Regression Equation:\n");
printf("y = %.4fx + %.4f\n", slope, intercept);
// Print prediction for sample x values
printf("\nPredicted y values:\n");
double test_x_values[] = {0, 2.5, 6, 10};
for (int i = 0; i < 4; i++) {
double predicted_y = predictY(slope, intercept, test_x_values[i]);
printf("When x = %.2f, y = %.4f\n", test_x_values[i], predicted_y);
}
return 0;
}
Compilez le programme :
gcc -o linear_regression linear_regression.c -lm
Exécutez le programme avec des points de données d'exemple :
./linear_regression
Exemple de sortie :
Enter x and y coordinates (enter -1 -1 to finish):
1 2
2 4
3 5
4 4
5 5
-1 -1
Linear Regression Equation:
y = 0.6000x + 2.2000
Predicted y values:
When x = 0.00, y = 2.2000
When x = 2.50, y = 3.7000
When x = 6.00, y = 5.8000
When x = 10.00, y = 8.2000
Points clés concernant l'affichage de l'équation de régression :
- Nous avons ajouté une fonction
predictY() pour calculer y pour tout x donné.
- La fonction principale affiche l'équation complète : y = mx + b.
- Nous démontrons la prédiction en montrant les valeurs de y pour différentes entrées de x.
- La sortie fournit une visualisation claire du modèle de régression linéaire.