pavanmutha commited on
Commit
d0adcb7
·
verified ·
1 Parent(s): fd6e2a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -127,7 +127,7 @@ def explainability(_):
127
 
128
  # Align number of columns
129
  if sv.shape[1] != X_test.shape[1]:
130
- X_test_trimmed = pd.DataFrame(X_test.values[:, :sv.shape[1]], columns=X_test.columns[:sv.shape[1]])
131
  else:
132
  X_test_trimmed = X_test
133
 
 
127
 
128
  # Align number of columns
129
  if sv.shape[1] != X_test.shape[1]:
130
+ X_test_trimmed = pd.DataFrame(X_test.values[:, :sv.shape[1]], columns=[f"Feature_{i}" for i in range(sv.shape[1])])
131
  else:
132
  X_test_trimmed = X_test
133