Балаганский Никита Николаевич
commited on
Commit
•
beee43b
1
Parent(s):
11aa723
fix softmax labels
Browse files
app.py
CHANGED
@@ -145,10 +145,8 @@ def main():
|
|
145 |
target_label_id = 1
|
146 |
act_type = "sigmoid"
|
147 |
else:
|
148 |
-
|
149 |
-
|
150 |
-
label_key = st.selectbox(ATTRIBUTE_LABEL[language], filtered_label2id.keys())
|
151 |
-
target_label_id = filtered_label2id[label_key]
|
152 |
act_type = "softmax"
|
153 |
st.write(WARNING_TEXT[language])
|
154 |
show_pos_alpha = st.checkbox("Show positive alphas", value=False)
|
|
|
145 |
target_label_id = 1
|
146 |
act_type = "sigmoid"
|
147 |
else:
|
148 |
+
label_key = st.selectbox(ATTRIBUTE_LABEL[language], ["Negative"])
|
149 |
+
target_label_id = 0
|
|
|
|
|
150 |
act_type = "softmax"
|
151 |
st.write(WARNING_TEXT[language])
|
152 |
show_pos_alpha = st.checkbox("Show positive alphas", value=False)
|