Spaces:
Sleeping
Sleeping
atodorov284 commited on
Commit ·
d3d65f2
1
Parent(s): 1b80238
Format code for flake8
Browse files
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]}")
|