J'ai formé mon modèle ML avec une régression linéaire en utilisant ces Comment puis-je obtenir une prédiction pour une seule ligne dans le jeu de données?
3 Réponses :
Vous pouvez facilement faire quelque chose comme ça- Donc, si vous voulez faire une inférence sur une seule ligne de l'ensemble de données, disons la ligne à la 2e Index. Vous feriez -
Modèles Scikit-Apprendre Tous ont un Il suffit de passer votre rangée comme une matrice et tout ira bien:
prédict Méthode que vous pouvez utiliser .
regressor.predict (x_val)
Puisque vous ne voulez que la première ligne, vous pouvez faire
first_row = X_test[0] //assuming X_test is where your test data is at y_pred = regressor.predict(first_row)
régresseur.predict (x_train)