File size: 568 Bytes
3be3f74 1e46458 3be3f74 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import streamlit as st
from sklearn import neighbors, datasets
with st.form(key='my_form'):
sLen = st.slider('sepal length (cm) ', 0.0, 10.0)
sWid = st.slider('sepal Width (cm) ', 0.0, 10.0)
pLen = st.slider('petal length (cm) ', 0.0, 10.0)
pWid = st.slider('petal Width (cm) ', 0.0, 10.0)
st.form_submit_button('predict')
iris = datasets.load_iris()
X,y = iris.data, iris.target
knn = neighbors.KNeighborsClassifier(n_neighbors=2) #k = 3,4,5,6
knn.fit(X,y)
predict = knn.predict([[sLen,sWid,pLen,pWid]])
st.write(iris.target_names[predict])
|