base

class thefittest.base.Tree(nodes: List[FunctionalNode | TerminalNode | EphemeralNode], n_args: List[int] | ndarray[Any, dtype[int64]] | None = None)

Bases: object

__init__(nodes: List[FunctionalNode | TerminalNode | EphemeralNode], n_args: List[int] | ndarray[Any, dtype[int64]] | None = None) → None

signature() → str

copy() → Tree

set_terminals(**kwargs: Any) → Tree

subtree_id(index: int) → Tuple[int, int]

subtree(index: int) → Tree

concat(index: int, other_tree: Tree) → Tree

get_common_region(other_trees: List | ndarray[Any, dtype[_ScalarType_co]]) → Tuple

get_args_id(index: int) → ndarray[Any, dtype[int64]]

get_levels(index: int) → ndarray[Any, dtype[int64]]

get_max_level() → int64

get_graph(keep_id: bool = False) → Dict

plot(ax: Any = None) → None

classmethod full_growing_method(uniset: UniversalSet, max_level: int) → Tree

classmethod growing_method(uniset: UniversalSet, max_level: int) → Tree

classmethod random_tree(uniset: UniversalSet, max_level: int) → Tree

class thefittest.base.FunctionalNode(value: Any, sign: str | None = None)

Bases: Node

__init__(value: Any, sign: str | None = None) → None

class thefittest.base.TerminalNode(value: Any, name: str)

Bases: Node

__init__(value: Any, name: str) → None

class thefittest.base.EphemeralNode(generator: Callable)

Bases: Node

__init__(generator: Callable) → None

class thefittest.base.UniversalSet(functional_set: Tuple[FunctionalNode, ...], terminal_set: Tuple[TerminalNode | EphemeralNode, ...])

Bases: object

__init__(functional_set: Tuple[FunctionalNode, ...], terminal_set: Tuple[TerminalNode | EphemeralNode, ...]) → None

class thefittest.base.Net(inputs: Set | None = None, hidden_layers: List | None = None, outputs: Set | None = None, connects: ndarray[Any, dtype[int64]] | None = None, weights: Tensor | None = None, activs: Dict[int, int] | None = None)

Bases: object

__init__(inputs: Set | None = None, hidden_layers: List | None = None, outputs: Set | None = None, connects: ndarray[Any, dtype[int64]] | None = None, weights: Tensor | None = None, activs: Dict[int, int] | None = None)

to(device: str | 'torch.device', dtype: 'torch.dtype' | None = None) → Net

cuda() → Net

cpu() → Net

copy() → Net

signature() → str

compile_torch(device: 'torch.device' | str | None = None) → Net

ensure_compiled() → None

forward(X: torch.Tensor, weights: 'torch.Tensor' | None = None, keep_weight_dim: bool = False, autocast_input: bool = True) → torch.Tensor

get_graph() → Dict

plot(ax=None) → None

thefittest.base.init_symbolic_regression_uniset(X: ndarray[Any, dtype[float64]], functional_set_names: Tuple[str, ...] = ('cos', 'sin', 'add', 'sub', 'mul', 'div'), ephemeral_node_generators: Tuple[Callable, ...] | None = None)