Advertisement
Guest User

Untitled

a guest
Jun 17th, 2019
60
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.37 KB | None | 0 0
  1. def permutation_importances(rf, X_train, y_train, metric):
  2. baseline = metric(rf, X_train, y_train)
  3. imp = []
  4. for col in X_train.columns:
  5. save = X_train[col].copy()
  6. X_train[col] = np.random.permutation(X_train[col])
  7. m = metric(rf, X_train, y_train)
  8. X_train[col] = save
  9. imp.append(baseline - m)
  10. return np.array(imp)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement