MatthiasPi commited on
Commit
5932aa1
ยท
1 Parent(s): f8bc968

Update algos/classification/svmclassifier.py

Browse files
algos/classification/svmclassifier.py CHANGED
@@ -3,10 +3,10 @@ import numpy as np
3
  from sklearn.pipeline import make_pipeline
4
  from sklearn.preprocessing import StandardScaler
5
  from sklearn.svm import SVC
6
- from types import NoneType
7
 
8
  def process(data):
9
- if type(data[0]) == NoneType or type(data[1]) == NoneType: # if either training or testing dataset is still missing
10
  st.info('Please Upload Data')
11
  return None
12
  if 'object' in list(data[0].dtypes) or 'object' in list(data[1].dtypes):
 
3
  from sklearn.pipeline import make_pipeline
4
  from sklearn.preprocessing import StandardScaler
5
  from sklearn.svm import SVC
6
+
7
 
8
  def process(data):
9
+ if data[0] == None or data[1] == None: # if either training or testing dataset is still missing
10
  st.info('Please Upload Data')
11
  return None
12
  if 'object' in list(data[0].dtypes) or 'object' in list(data[1].dtypes):