louiecerv commited on
Commit
85afa33
·
1 Parent(s): 8eda214
Files changed (2) hide show
  1. requirements.txt +2 -1
  2. src/streamlit_app.py +2 -0
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  altair
2
  pandas
3
- streamlit
 
 
1
  altair
2
  pandas
3
+ streamlit
4
+ scikit-learn
src/streamlit_app.py CHANGED
@@ -1,5 +1,7 @@
1
  import streamlit as st
2
  import pickle
 
 
3
 
4
  rf_pickle = open('model/random_forest_penguin.pickle', 'rb')
5
  map_pickle = open('model/uniques_data.pickle', 'rb')
 
1
  import streamlit as st
2
  import pickle
3
+ from sklearn.model_selection import train_test_split
4
+ import pickle
5
 
6
  rf_pickle = open('model/random_forest_penguin.pickle', 'rb')
7
  map_pickle = open('model/uniques_data.pickle', 'rb')