MatthiasPi commited on
Commit
f8bc968
ยท
1 Parent(s): 06d38ce

Update algos/classification/nnclassifier.py

Browse files
algos/classification/nnclassifier.py CHANGED
@@ -1,10 +1,10 @@
1
  import streamlit as st
2
  from sklearn.neural_network import MLPClassifier
3
  import pandas as pd
4
- from types import NoneType
5
 
6
  def process(data):
7
- if type(data[0]) == NoneType or type(data[1]) == NoneType: # if either training or testing dataset is still missing
8
  st.info('Please Upload Data')
9
  return None
10
  if 'object' in list(data[0].dtypes) or 'object' in list(data[1].dtypes):
 
1
  import streamlit as st
2
  from sklearn.neural_network import MLPClassifier
3
  import pandas as pd
4
+
5
 
6
  def process(data):
7
+ if data[0] == None or data[1] == None: # if either training or testing dataset is still missing
8
  st.info('Please Upload Data')
9
  return None
10
  if 'object' in list(data[0].dtypes) or 'object' in list(data[1].dtypes):