Scribd
Upload
13 views

Unit-3 Machine Learning Model with FastAPI for Iris Dataset

i am sagar kuamr

Uploaded by

Shubham Singh Rajput
Copyright
© © All Rights Reserved
Available Formats
Download as PDF, TXT or read online on Scribd
13 views

Unit-3 Machine Learning Model with FastAPI for Iris Dataset

i am sagar kuamr

Uploaded by

Shubham Singh Rajput
Copyright
© © All Rights Reserved
Available Formats
Download as PDF, TXT or read online on Scribd
You are on page 1/ 3

Machine Learning Model with FastAPI for Iris Dataset

# File 1: MLDemo.py → This file contains ML Training and its model using pickle

# Importing required libraries


import pandas as pd
from sklearn.datasets import load_iris
from sklearn.ensemble import RandomForestClassifier
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
import pickle

# Load the Iris dataset


iris = load_iris()
X = iris['data']
y = iris['target']

# Split the data into training and testing sets


X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)

# Initialize and train the Random Forest Classifier


model = RandomForestClassifier(n_estimators=100, random_state=42)
model.fit(X_train, y_train)

# Evaluate the model


y_pred = model.predict(X_test)
accuracy = accuracy_score(y_test, y_pred)
print(f"Model Accuracy: {accuracy * 100}%")

# Save the model to a file using pickle


with open('iris_model.pkl', 'wb') as f:
pickle.dump(model, f)
# File 2 : MLFastAPI.py → This file contains the loading of .pkl model created above and creating its
API

import pickle
from fastapi import FastAPI
from pydantic import BaseModel

# Define the request body for input data


class IrisRequest(BaseModel):
sepal_length: float
sepal_width: float
petal_length: float
petal_width: float

# Load the pretrained model


with open("iris_model.pkl", "rb") as f:
model = pickle.load(f)

# Load the iris dataset to access target names


from sklearn.datasets import load_iris

iris = load_iris()
target_names = iris.target_names # ['setosa', 'versicolor', 'virginica']

# Create FastAPI instance


app = FastAPI()

# Root endpoint for testing


@app.get("/")
def root():
return {"message": "Welcome to the Iris Prediction API"}

# Prediction endpoint
@app.post("/predict/")
def predict_iris(data: IrisRequest):
# Prepare input data as a list of features
features = [[
data.sepal_length,
data.sepal_width,
data.petal_length,
data.petal_width
]]
# Predict the class
prediction = model.predict(features)
predicted_class = target_names[prediction[0]] # Map class number to class name

return {"predicted_class": predicted_class}

You might also like