diff options
Diffstat (limited to 'predict.py')
| -rw-r--r-- | predict.py | 9 |
1 files changed, 6 insertions, 3 deletions
@@ -4,6 +4,7 @@ import pandas as pd # Load the model mileage_model = joblib.load('mileage_predictor.pkl') price_model = joblib.load('price_predictor.pkl') +year_model = joblib.load('year_predictor.pkl') # Prepare input data for prediction def prepare_input(data_dict): @@ -20,12 +21,13 @@ def predict(input_data): prepared_data = prepare_input(input_data) mileage = mileage_model.predict(prepared_data)[0] price = price_model.predict(prepared_data)[0] + year = year_model.predict(prepared_data)[0] - return mileage, price + return mileage, price, int(year) # Sample data for prediction data = { - 'Year': 2018, + 'Year': 2022, 'Kilometers_Driven': 30000, 'Fuel_Type': 'Petrol', 'Transmission': 'Manual', @@ -38,7 +40,8 @@ data = { # Make prediction -predicted_mileage, predicted_price = predict(data) +predicted_mileage, predicted_price, predicted_year = predict(data) print(f"Predicted Mileage (Km/L): {predicted_mileage:.2f}") print(f"Predicted Price: ₹{predicted_price:,.2f} Lakhs") +print(f"Predicted Year: {predicted_year}") |