search

MilesCranmer / PySR

High-Performance Symbolic Regression in Python and Julia
https://ai.damtp.cam.ac.uk/pysr
Apache License 2.0
2.44k stars 217 forks source link

Per-variable custom complexities #649

Closed MilesCranmer closed 5 months ago

MilesCranmer commented 5 months ago

This allows you to specify different complexities for each variable:

model.fit(X, y, complexity_of_variables=[1, 3])

will run a search with feature 1 having complexity 1 and feature 2 with complexity 3. This also updates the backend with various improvements.