saifhmb commited on
Commit
098e9ff
·
unverified ·
1 Parent(s): f4917b0

dataset['MARITAL'] = dataset['MARITAL'].replace(['married', 'single', 'divsepwid'], [0, 1, 2], inplace = True)

Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -23,11 +23,12 @@ dataset_name = "saifhmb/CreditCardRisk"
23
  dataset = load_dataset(dataset_name, split = 'train')
24
  dataset = pd.DataFrame(dataset)
25
  dataset['MARITAL'] = dataset['MARITAL'].str.replace(' ', '')
 
26
  X = dataset.iloc[:, :-1].values
27
  y = dataset.iloc[:, -1].values
28
 
29
  # Encoding the Independent Variables
30
- ct = ColumnTransformer(transformers = [('encoder', OneHotEncoder(sparse_output=False, max_categories = 3), [2, 3, 6, 7])], remainder = 'passthrough')
31
  X = np.array(ct.fit_transform(X))
32
  #X= X.astype('int')
33
 
 
23
  dataset = load_dataset(dataset_name, split = 'train')
24
  dataset = pd.DataFrame(dataset)
25
  dataset['MARITAL'] = dataset['MARITAL'].str.replace(' ', '')
26
+ dataset['MARITAL'] = dataset['MARITAL'].replace(['married', 'single', 'divsepwid'], [0, 1, 2], inplace = True)
27
  X = dataset.iloc[:, :-1].values
28
  y = dataset.iloc[:, -1].values
29
 
30
  # Encoding the Independent Variables
31
+ ct = ColumnTransformer(transformers = [('encoder', OneHotEncoder(sparse_output=False), [2, 3, 6, 7])], remainder = 'passthrough')
32
  X = np.array(ct.fit_transform(X))
33
  #X= X.astype('int')
34