pykan

app.py

@@ -69,6 +69,7 @@ def show_kan_prediction(model, device, samples, placeholder, phase_name):
     
     # 添加采样点
     if samples is not None:
+        samples = samples.cpu().numpy() if torch.is_tensor(samples) else samples
         fig_kan.add_trace(
             go.Scatter(
                 x=samples[:, 0], y=samples[:, 1],
@@ -320,7 +321,9 @@ def train_kan(samples, gmm_dataset, device='cuda'):
     with st.spinner("符号简化中..."):
         # model = model.prune()
         # progress_container.info("网络剪枝完成")
-        model.auto_symbolic()
+        lib = ['x','x^2','x^3','x^4','exp','log','sqrt','tanh','sin','abs']
+        model.auto_symbolic(lib=lib)
+        # model.auto_symbolic()
         progress_container.info("符号简化完成")
 
     with st.spinner("参数调整中..."):

requirements.txt

@@ -8,5 +8,5 @@ ipython>=8.0.0
 ipywidgets>=7.0.0
 nbformat>=5.0.0
 sympy>=1.8
-pykan
+git+https://github.com/2catycm/pykan.git
 matplotlib