deeperwin.model.BaselineOrbitals

class deeperwin.model.BaselineOrbitals(*args, **kwargs)

Bases: haiku._src.module.Module

__init__(config: deeperwin.configuration.BaselineOrbitalsConfig, physical_config: deeperwin.configuration.PhysicalConfig, mlp_config: deeperwin.configuration.MLPConfig, n_dets, full_det, name=None)

Initializes the current module with the given name.

Subclasses should call this constructor before creating other modules or variables such that those modules are named correctly.

Parameters

name – An optional string name for the class. Must be a valid Python identifier. If name is not provided then the class name for the current instance is converted to lower_snake_case and used instead.

Methods

__init__(config, physical_config, ...[, name])	Initializes the current module with the given name.
apply_backflow_shift(diff_dist, embeddings)