Skip to content

Instantly share code, notes, and snippets.

@mkhorasani
Last active September 7, 2021 18:46
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save mkhorasani/a9c42799aa4e4be708629c822bb82c82 to your computer and use it in GitHub Desktop.
Save mkhorasani/a9c42799aa4e4be708629c822bb82c82 to your computer and use it in GitHub Desktop.
def download(df,filename): # Downloading DataFrame
csv = df.to_csv(index=False)
b64 = base64.b64encode(csv.encode()).decode()
href = (f'<a href="data:file/csv;base64,{b64}" download="%s.csv">Download csv file</a>' % (filename))
return href
st.sidebar.subheader('Test Dataset')
status_test, df_test = file_upload('Please upload a test dataset')
if status_test == True:
try:
st.title('Testing')
X_test_test = df_test[feature_cols]
y_pred_test = logreg.predict(X_test_test)
X_pred = df_test.copy()
X_pred[label_col] = y_pred_test
X_pred = X_pred.sort_index()
st.subheader('Predicted Labels')
st.write(X_pred)
#st.write(X_pred[label_col].value_counts())
st.markdown(download(X_pred,'Logistic Regression Classifier - Predicted Labels'), unsafe_allow_html=True)
except:
st.warning('Please upload a test dataset with the same feature set as the training dataset')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment