1

pages/2_TECT_IDXYZ.py

@@ -34,6 +34,14 @@ def GetListOf_XYZV_ToTrainClassifier(repo):
     lstDfOriginal = [my_stm.df_process_v_column(df) for df in dictTrainThreeDataframes.values()]
     return lstDfOriginal
 
+@st.cache_data
+def GetCachedClassifier(lstDfOriginal, nHystorySteps):
+    classifier_object, df_train_with_predict, time_elapsed = my_stm.GetClassifier(lstDfOriginal, nHystorySteps)
+    st.session_state.df_train_with_predict = df_train_with_predict
+    print(f"GetCachedClassifier() : {time_elapsed=}")
+    return classifier_object
+
+
 def ReRun() :
     try: st.rerun()
     except: pass
@@ -128,7 +136,8 @@ if selectedFile is not None:
         #col1.dataframe(dfShow, height=400)
         pack_size = list(set(dgID.apply(len)))[0]
         lstDfOriginal = GetListOf_XYZV_ToTrainClassifier(REPO)
-        classifier_object, df_train_with_predict, time_elapsed = my_stm.GetClassifier(lstDfOriginal, pack_size-1)
+        #classifier_object, df_train_with_predict, time_elapsed = my_stm.GetClassifier(lstDfOriginal, pack_size-1)
+        classifier_object = GetCachedClassifier(lstDfOriginal, pack_size-1)
         col2.popover(type(classifier_object).__name__).write(type(classifier_object))
         # прогноз на обучающей выборке
         columns_xyzv = [c for c in df_train_with_predict.columns if "Vis" in c] + [c for c in df_train_with_predict.columns if c[0] in "XYZ"]