sgbaird commited on
Commit
1e10b8f
·
1 Parent(s): 09b30cb

basic multioutput example with reqs

Browse files
Files changed (2) hide show
  1. app.py +32 -0
  2. requirements.txt +3 -0
app.py ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import pandas as pd
3
+
4
+ from sklearn.multioutput import MultiOutputRegressor
5
+
6
+ from sklearn.datasets import load_linnerud
7
+ from sklearn.multioutput import MultiOutputRegressor
8
+ from sklearn.linear_model import Ridge
9
+
10
+ X, y = load_linnerud(return_X_y=True, as_frame=True)
11
+ regr = MultiOutputRegressor(Ridge(random_state=123)).fit(X, y)
12
+ # regr.predict(X.iloc[[0]])
13
+
14
+
15
+ iface = gr.Interface(
16
+ fn=regr.predict,
17
+ inputs=gr.Dataframe(
18
+ value=X.head(1),
19
+ headers=list(X.columns),
20
+ col_count=(X.shape[1], "fixed"),
21
+ row_count=(1, "dynamic"),
22
+ datatype=X.dtypes.apply(str).replace("float64", "number").values.tolist(),
23
+ ),
24
+ outputs=gr.Numpy(
25
+ value=regr.predict(X.head(1)),
26
+ headers=list(y.columns),
27
+ col_count=(y.shape[1], "fixed"),
28
+ datatype=y.dtypes.apply(str).replace("float64", "number").values.tolist(),
29
+ min_width=50,
30
+ ),
31
+ )
32
+ iface.launch()
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ numpy
2
+ gradio
3
+ scikit-learn