Bayhaqy's picture
Update app.py
bfdebbf
from streamlit_pandas_profiling import st_profile_report
from ydata_profiling import ProfileReport
import streamlit as st
import pandas as pd
import numpy as np
from prediction import predict
from function import filter_dataframe
from sklearn.datasets import load_iris
from ydata_profiling.utils.cache import cache_file
st.set_page_config(layout="wide")
st.title('Iris Flowers - Classification')
st.caption('Created by Bayhaqy')
st.markdown('Classify iris flowers into \
setosa, versicolor, virginica')
st.image('https://machinelearninghd.com/wp-content/uploads/2021/03/iris-dataset.png')
st.image('https://www.integratedots.com/wp-content/uploads/2019/06/iris_petal-sepal-e1560211020463.png')
st.write("---")
# Load Dataset
@st.cache_data
def load_data(url):
df = pd.read_csv(url)
return df
iris = cache_file(
'Iris.csv',
'https://raw.githubusercontent.com/bayhaqy/Classification-Iris-Prediction/main/Iris.csv',
)
df = load_data(iris)
if st.checkbox('Open Iris Dataset'):
fd = filter_dataframe(df)
st.dataframe(fd, use_container_width=True)
st.write("---")
if st.checkbox('Open EDA Report'):
pr = ProfileReport(df)
st_profile_report(pr)
st.write("---")
st.header('Plant Features')
col1, col2 = st.columns(2)
with col1:
st.text('Sepal Size')
sepal_l = st.slider('Sepal lenght (cm)', 1.0, 8.0, 0.5)
sepal_w = st.slider('Sepal width (cm)', 2.0, 4.4, 0.5)
with col2:
st.text('Pepal Size')
petal_l = st.slider('Petal lenght (cm)', 1.0, 7.0, 0.5)
petal_w = st.slider('Petal width (cm)', 0.1, 2.5, 0.5)
if st.button('Predict type of Iris'):
result = predict(np.array([[sepal_l, sepal_w, petal_l, petal_w]]))
st.text(result[0])