Tweet / app.py
ErenKontas's picture
Upload 6 files
cadb7ff verified
raw
history blame
1.47 kB
import pandas as pd
import streamlit as st
import joblib
from sklearn.preprocessing import StandardScaler, OneHotEncoder
from sklearn.model_selection import train_test_split
from sklearn.compose import ColumnTransformer
# Veriyi yükleme ve sütun isimlerini güncelleme
df = pd.read_csv('train.csv')
# Bağımlı ve bağımsız değişkenlerin seçimi
x = df.drop(['essay_id', 'text'], axis=1)
y = df[['text']]
# Eğitim ve test verilerini ayırma
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.20, random_state=42)
# Ön işleme (StandardScaler ve OneHotEncoder)
preprocessor = ColumnTransformer(
transformers=[
('num', StandardScaler(), ['feeling']),
('cat', OneHotEncoder(), ['text'])
]
)
# Streamlit uygulaması
def rings_pred(feeling, text):
input_data = pd.DataFrame({
'text': [text],
'feeling': [feeling]
})
input_data_transformed = preprocessor.fit_transform(input_data)
model = joblib.load('Tweet.pkl')
prediction = model.predict(input_data_transformed)
return float(prediction[0])
st.title("Abalone Veri seti ile Yaş Tahmini Regresyon Modeli")
st.write("Veri Gir")
text = st.selectbox('text', df['text'].unique())
feeling = st.selectbox('feeling', df['feeling'].unique())
if st.button('Predict'):
rings = rings_pred(text,feeling)
st.write(f'The predicted rings is: {rings:.2f}')