search

MilesCranmer / PySR

High-Performance Symbolic Regression in Python and Julia
https://astroautomata.com/PySR
Apache License 2.0
2.32k stars 211 forks source link

Per-variable custom complexities #649

Closed MilesCranmer closed 3 months ago

MilesCranmer commented 3 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.