pyvene.models.mlp.modelings_intervenable_mlp.mlp_type_to_dimension_mapping

mlp_type_to_dimension_mapping = {'block_input': ('h_dim',), 'block_output': ('h_dim',), 'mlp_activation': ('h_dim',)}

mlp model with classification head