Amazing-Python-Scripts
43 строки · 1.2 Кб
1import streamlit as st2import numpy as np3import pandas as pd4from sklearn.cluster import KMeans5from sklearn.model_selection import train_test_split6from sklearn.linear_model import LogisticRegression7from sklearn.metrics import classification_report8data = pd.read_excel("data.xlsx")9y = data['label']10X = data.drop(["label"], axis=1)11X_train, X_test, y_train, y_test = train_test_split(12X, y, test_size=0.3, random_state=0)13lr = LogisticRegression()14lr.fit(X_train, y_train)15
16
17def predict_crop(input_data):18crop_label = lr.predict(input_data)19return crop_label[0]20
21
22def main():23st.title("Agriculture Optimisation App")24st.write("Enter the parameter values to predict the crop label:")25
26n = st.number_input("N", value=0.0)27p = st.number_input("P", value=0.0)28k = st.number_input("K", value=0.0)29temperature = st.number_input("Temperature", value=0.0)30humidity = st.number_input("Humidity", value=0.0)31ph = st.number_input("pH", value=0.0)32rainfall = st.number_input("Rainfall", value=0.0)33
34input_data = np.array([[n, p, k, temperature, humidity, ph, rainfall]])35
36crop_label = predict_crop(input_data)37
38st.subheader("Predicted Crop Label:")39st.write(crop_label)40
41
42if __name__ == '__main__':43main()44