Views
No views yet
| File | Description |
|---|---|
yield_pipeline.pkl | Full sklearn pipeline (preprocessor + model) |
rf_model.pkl | Random Forest base model artifact |
le_crop.pkl | Label encoder for crop names |
le_state.pkl | Label encoder for Indian states |
le_dist.pkl | Label encoder for districts |
cleaned_dataset.csv | Training dataset reference |
Crop, State, District (categorical)Area (ha), Temperature, Humidity, Rainfall, N, P, K (numeric)predicted_yield in kg/ha1import joblib
2import pandas as pd
3
4model = joblib.load("yield_pipeline.pkl")
5
6# Inputs: Crop, State, District, Area, Temp, Humid, Rain, N, P, K
7data = pd.DataFrame([[
8 "Rice", "West Bengal", "PURBA BARDHAMAN",
9 500, 25.5, 80, 1500, 80, 40, 40
10]], columns=[
11 'Crop', 'State', 'District', 'Area', 'Temperature',
12 'Humidity', 'Rainfall', 'N', 'P', 'K'
13])
14
15prediction = model.predict(data)
16print(f"Predicted Yield: {prediction[0]:.2f} kg/ha")