atodorov284 commited on
Commit
d3d65f2
·
1 Parent(s): 1b80238

Format code for flake8

Browse files
Files changed (1) hide show
  1. extra_scripts/shap_values.py +1 -1
extra_scripts/shap_values.py CHANGED
@@ -51,7 +51,7 @@ if __name__ == "__main__":
51
  """output_features = ['NO2 - Day 1', 'O3 - Day 1', 'NO2 - Day 2', 'O3 - Day 2', 'NO2 - Day 3', 'O3 - Day 3']
52
 
53
  shap_values = explainer.shap_values(x_test)
54
- n_outputs = shap_values.shape[2]
55
 
56
  for i in range(n_outputs):
57
  print(f"Generating summary plot for {output_features[i]}")
 
51
  """output_features = ['NO2 - Day 1', 'O3 - Day 1', 'NO2 - Day 2', 'O3 - Day 2', 'NO2 - Day 3', 'O3 - Day 3']
52
 
53
  shap_values = explainer.shap_values(x_test)
54
+ n_outputs = shap_values.shape[2]
55
 
56
  for i in range(n_outputs):
57
  print(f"Generating summary plot for {output_features[i]}")