带整数参数约束的符号回归实现求助

我正在做一个符号回归项目，目前用scipy的curve_fit来拟合线性表达式的参数（对应sr.py文件第85行的fit函数），但处理非线性表达式时效率极低——我在tests/test_sr.py里写了相关测试用例，现在只能先注释掉，根本没法正常运行。

为了简化问题，我打算给参数加上约束，比如强制参数为整数。之前尝试用DEAP来实现，但折腾了好一阵都没成功，估计是我用法出了问题。

想请教下社区的各位：有没有合适的思路或者工具能帮我实现带整数参数约束的符号回归？如果有使用DEAP处理这类需求的经验，也麻烦指点下我可能踩了哪些坑，非常感谢！

内容来源于stack exchange