Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -105,16 +105,13 @@ class EuroSATClassifier:
|
|
| 105 |
return None, None, error_msg
|
| 106 |
|
| 107 |
def create_confidence_plot(self, results):
|
| 108 |
-
"""Create a confidence plot using Plotly"""
|
| 109 |
classes = list(results.keys())
|
| 110 |
confidences = [results[cls] * 100 for cls in classes]
|
| 111 |
-
|
| 112 |
-
colors
|
| 113 |
-
|
| 114 |
-
|
| 115 |
-
]
|
| 116 |
-
|
| 117 |
-
|
| 118 |
fig = go.Figure(data=[
|
| 119 |
go.Bar(
|
| 120 |
x=confidences,
|
|
@@ -126,35 +123,31 @@ class EuroSATClassifier:
|
|
| 126 |
textfont=dict(color='white', size=12),
|
| 127 |
)
|
| 128 |
])
|
| 129 |
-
|
| 130 |
fig.update_layout(
|
| 131 |
-
title=
|
| 132 |
-
'text': "🎯 Classification Confidence Scores",
|
| 133 |
-
'x': 0.5,
|
| 134 |
-
'xanchor': 'center',
|
| 135 |
-
'font': {'size': 16, 'color': '#2C3E50'}
|
| 136 |
-
},
|
| 137 |
xaxis_title="Confidence (%)",
|
| 138 |
yaxis_title="Land Cover Classes",
|
| 139 |
height=500,
|
| 140 |
-
margin=dict(l=10, r=10, t=
|
| 141 |
-
plot_bgcolor='
|
| 142 |
paper_bgcolor='white',
|
| 143 |
-
font=dict(family="Arial
|
| 144 |
xaxis=dict(
|
| 145 |
-
gridcolor='rgba(
|
| 146 |
showgrid=True,
|
| 147 |
range=[0, 100]
|
| 148 |
),
|
| 149 |
yaxis=dict(
|
| 150 |
-
gridcolor='rgba(
|
| 151 |
showgrid=True,
|
| 152 |
-
autorange="reversed"
|
| 153 |
)
|
| 154 |
)
|
| 155 |
-
|
| 156 |
return fig
|
| 157 |
|
|
|
|
| 158 |
# Initialize the classifier
|
| 159 |
classifier = EuroSATClassifier()
|
| 160 |
|
|
|
|
| 105 |
return None, None, error_msg
|
| 106 |
|
| 107 |
def create_confidence_plot(self, results):
|
| 108 |
+
"""Create a clean confidence plot using Plotly"""
|
| 109 |
classes = list(results.keys())
|
| 110 |
confidences = [results[cls] * 100 for cls in classes]
|
| 111 |
+
|
| 112 |
+
# Use consistent solid colors (green for top, blue for others)
|
| 113 |
+
colors = ['#2E8B57' if i == 0 else '#4682B4' for i in range(len(classes))]
|
| 114 |
+
|
|
|
|
|
|
|
|
|
|
| 115 |
fig = go.Figure(data=[
|
| 116 |
go.Bar(
|
| 117 |
x=confidences,
|
|
|
|
| 123 |
textfont=dict(color='white', size=12),
|
| 124 |
)
|
| 125 |
])
|
| 126 |
+
|
| 127 |
fig.update_layout(
|
| 128 |
+
title="🎯 Classification Confidence Scores",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 129 |
xaxis_title="Confidence (%)",
|
| 130 |
yaxis_title="Land Cover Classes",
|
| 131 |
height=500,
|
| 132 |
+
margin=dict(l=10, r=10, t=40, b=10),
|
| 133 |
+
plot_bgcolor='white',
|
| 134 |
paper_bgcolor='white',
|
| 135 |
+
font=dict(family="Arial", size=12, color="#333"),
|
| 136 |
xaxis=dict(
|
| 137 |
+
gridcolor='rgba(0,0,0,0.05)',
|
| 138 |
showgrid=True,
|
| 139 |
range=[0, 100]
|
| 140 |
),
|
| 141 |
yaxis=dict(
|
| 142 |
+
gridcolor='rgba(0,0,0,0.05)',
|
| 143 |
showgrid=True,
|
| 144 |
+
autorange="reversed"
|
| 145 |
)
|
| 146 |
)
|
| 147 |
+
|
| 148 |
return fig
|
| 149 |
|
| 150 |
+
|
| 151 |
# Initialize the classifier
|
| 152 |
classifier = EuroSATClassifier()
|
| 153 |
|