Corey Morris
commited on
Commit
·
b94ee8f
1
Parent(s):
643f05d
for the custom charts, remove columns with NaN values
Browse files
app.py
CHANGED
|
@@ -49,6 +49,9 @@ def create_plot(df, arc_column, moral_column, models=None):
|
|
| 49 |
if models is not None:
|
| 50 |
df = df[df.index.isin(models)]
|
| 51 |
|
|
|
|
|
|
|
|
|
|
| 52 |
plot_data = pd.DataFrame({
|
| 53 |
'Model': df.index,
|
| 54 |
arc_column: df[arc_column],
|
|
|
|
| 49 |
if models is not None:
|
| 50 |
df = df[df.index.isin(models)]
|
| 51 |
|
| 52 |
+
# remove rows with NaN values
|
| 53 |
+
df = df.dropna(subset=[arc_column, moral_column])
|
| 54 |
+
|
| 55 |
plot_data = pd.DataFrame({
|
| 56 |
'Model': df.index,
|
| 57 |
arc_column: df[arc_column],
|