Update app.py

app.py

@@ -0,0 +1,65 @@
+import pandas as pd
+import numpy as np
+import sklearn
+import gradio as gr
+from sklearn import preprocessing
+from sklearn.model_selection import train_test_split
+from sklearn.ensemble import RandomForestClassifier
+from sklearn.metrics import accuracy_score
+
+data = pd.read_csv('https://raw.githubusercontent.com/gradio-app/titanic/master/train.csv')
+data.head()
+
+def encode_ages(df): # Binning ages 
+    df.Age = df.Age.fillna(-0.5)
+    bins = (-1, 0, 5, 12, 18, 25, 35, 60, 120)
+    categories = pd.cut(df.Age, bins, labels=False)
+    df.Age = categories
+    return df
+
+def encode_fares(df): # Binning fares
+    df.Fare = df.Fare.fillna(-0.5)
+    bins = (-1, 0, 8, 15, 31, 1000)
+    categories = pd.cut(df.Fare, bins, labels=False)
+    df.Fare = categories
+    return df
+
+def encode_sex(df):
+    mapping = {"male": 0, "female": 1}
+    return df.replace({'Sex': mapping})
+
+def transform_features(df):
+    df = encode_ages(df)
+    df = encode_fares(df)
+    df = encode_sex(df)
+    return df
+
+train = data[['PassengerId', 'Fare', 'Age', 'Sex', 'Survived']]
+train = transform_features(train)
+train.head()
+
+
+X_all = train.drop(['Survived', 'PassengerId'], axis=1)
+y_all = train['Survived']
+
+num_test = 0.20
+X_train, X_test, y_train, y_test = train_test_split(X_all, y_all, test_size=num_test, random_state=23)
+
+clf = RandomForestClassifier()
+clf.fit(X_train, y_train)
+predictions = clf.predict(X_test)
+
+def predict_survival(sex, age, fare):
+    df = pd.DataFrame.from_dict({'Sex': [sex], 'Age': [age], 'Fare': [fare]})
+    df = encode_sex(df)
+    df = encode_fares(df)
+    df = encode_ages(df)
+    pred = clf.predict_proba(df)[0]
+    return {'Muere': float(pred[0]), 'Sobrevive': float(pred[1])}
+    
+sex = gr.inputs.Radio(['Mujer', 'Varon'], label="Genero")
+age = gr.inputs.Slider(minimum=0, maximum=120, default=22, label="Edad")
+fare = gr.inputs.Slider(minimum=0, maximum=200, default=100, label="Tarifa (british pounds)")
+
+gr.Interface(predict_survival, [sex, age, fare], "label", live=True, thumbnail="https://raw.githubusercontent.com/gradio-app/hub-titanic/master/thumbnail.png", analytics_enabled=False,
+    title="Sobreviviendo al Titanic", description="Cual es la probabilidad de sobrevivir al famoso accidente del titanic? Depende de indicadores demograficos como se muestra a continuacion.").launch();