kdnv update

pages/kudinov_films.py

@@ -6,6 +6,7 @@ import json
 import transformers
 from torch import nn
 import joblib
+import pandas as pd
 
 
 autotoken = transformers.AutoTokenizer.from_pretrained(
@@ -92,3 +93,34 @@ if submit:
     with col3:
         st.metric(label="Logistic Regression Prediction", value=ans_lr)
 
+data = {
+    "Model Setup": [
+        "logreg tf-idf",
+        "rf tf-idf",
+        "catboost tf-idf",
+        "svc tf-idf",
+        "logreg bow",
+        "rf bow",
+        "catboost bow",
+        "svc bow",
+        "word2vec + LSTM + богдан",
+        "bert + linear"
+    ],
+    "F1 Score": [
+        "0.666", "0.608", "0.654", "0.666",
+        "0.648", "0.563", "0.619", "0.646",
+        "0.653", "0.660"
+    ]
+}
+
+df = pd.DataFrame(data)
+
+st.title("Результаты обучения моделей")
+st.table(df)
+
+
+st.image('images/lstm.png')
+st.image('images/output.png')
+st.image('images/logreg.png')
+
+