spark.nn.brain

Classes

BrainMeta	Brain metaclass.
BrainConfig	Configuration class for Brain's.
Brain	Abstract brain model.

Module Contents

class spark.nn.brain.BrainMeta

Bases: spark.core.module.SparkMeta

Brain metaclass.

class spark.nn.brain.BrainConfig(**kwargs)

Bases: spark.core.config.BaseSparkConfig

Configuration class for Brain’s.

input_map: dict[str, spark.core.specs.InputSpec]

output_map: dict[str, dict]

modules_map: dict[str, spark.core.specs.ModuleSpecs]

validate()

Validates all fields in the configuration class.

Return type:

None

class spark.nn.brain.Brain(config=None, **kwargs)

Bases: spark.core.module.SparkModule

Abstract brain model. This is more a convenience class used to synchronize data more easily.

Parameters:

config (BrainConfig)

config: BrainConfig

resolve_initialization_order()

Resolves the initialization order of the modules.

build(input_specs)

Build method.

Parameters:

input_specs (dict[str, spark.core.specs.InputSpec])

reset()

Resets all the modules to its initial state.

__call__(**inputs)

Update brain’s states.

Parameters:

inputs (spark.core.payloads.SparkPayload)

Return type:

tuple[spark.core.payloads.SparkPayload]

get_spikes_from_cache()

Collect the brain’s spikes.

Return type:

dict