Index

A

access_note (mtlearn.data.DatasetSpec attribute) Attribute (class in mtlearn.morphology)

AttributeFilterDataset (class in mtlearn.datasets) AttributeFilters (class in mtlearn.morphology)

B

build_dataloader_cached() (mtlearn.layers.ConnectedFilterPreprocessingLayer method) (mtlearn.layers.ConnectedFilterPreprocessingLayerLegacy method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithCPUTreeTraversal method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithExplicitJacobian method)

build_dataloader_cached_fixed_stats() (mtlearn.layers.ConnectedFilterPreprocessingLayer method) build_tree() (in module mtlearn.morphology)

C

collect_cfp_configs() (in module mtlearn.layers) compute_attributes() (in module mtlearn.morphology) compute_single_attribute() (in module mtlearn.morphology) ConnectedFilterPreprocessingLayer (class in mtlearn.layers) ConnectedFilterPreprocessingLayerLegacy (class in mtlearn.layers) ConnectedFilterPreprocessingLayerWithCPUTreeTraversal (class in mtlearn.layers)

ConnectedFilterPreprocessingLayerWithExplicitJacobian (class in mtlearn.layers) ConnectedFilterPreprocessingTreeTensors (class in mtlearn) ConnectedFilterPreprocessingTreeTraversal (class in mtlearn) create_attribute_filter() (in module mtlearn.morphology) create_max_tree() (in module mtlearn.morphology) create_min_tree() (in module mtlearn.morphology) create_tree_of_shapes() (in module mtlearn.morphology)

D

dataset_path() (in module mtlearn.data) DatasetSpec (class in mtlearn.data) default_data_dir() (in module mtlearn.data) describe_all_attributes() (in module mtlearn.morphology)

describe_attribute() (in module mtlearn.morphology) description (mtlearn.data.DatasetSpec attribute) download_by_default (mtlearn.data.DatasetSpec attribute) download_file() (in module mtlearn.data) dropbox_download_url() (in module mtlearn.data)

E

ensure_dataset() (in module mtlearn.data) expand_attribute_group() (in module mtlearn.morphology)

export_params() (mtlearn.layers.ConnectedFilterPreprocessingLayer method) extracted_content_root() (in module mtlearn.data)

F

format_size() (in module mtlearn.data) forward() (mtlearn.layers.ConnectedFilterPreprocessingLayer method) (mtlearn.layers.ConnectedFilterPreprocessingLayerLegacy method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithCPUTreeTraversal method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithExplicitJacobian method)

freeze_ds_stats() (mtlearn.layers.ConnectedFilterPreprocessingLayer method) (mtlearn.layers.ConnectedFilterPreprocessingLayerLegacy method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithCPUTreeTraversal method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithExplicitJacobian method) from_config() (mtlearn.layers.ConnectedFilterPreprocessingLayer class method)

G

get_config() (mtlearn.layers.ConnectedFilterPreprocessingLayer method) get_extra_state() (mtlearn.layers.ConnectedFilterPreprocessingLayer method) get_params() (mtlearn.layers.ConnectedFilterPreprocessingLayer method) (mtlearn.layers.ConnectedFilterPreprocessingLayerLegacy method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithCPUTreeTraversal method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithExplicitJacobian method)

get_weight_contract() (mtlearn.layers.ConnectedFilterPreprocessingLayer method)

H

has_existing_files() (in module mtlearn.data)

I

init_identity_bias_zero() (mtlearn.layers.ConnectedFilterPreprocessingLayer method) (mtlearn.layers.ConnectedFilterPreprocessingLayerLegacy method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithCPUTreeTraversal method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithExplicitJacobian method) init_identity_with_bias() (mtlearn.layers.ConnectedFilterPreprocessingLayer method) (mtlearn.layers.ConnectedFilterPreprocessingLayerLegacy method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithCPUTreeTraversal method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithExplicitJacobian method)

inspect_training_sample() (mtlearn.layers.ConnectedFilterPreprocessingLayer method) (mtlearn.layers.ConnectedFilterPreprocessingLayerLegacy method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithCPUTreeTraversal method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithExplicitJacobian method) is_tree() (in module mtlearn.morphology)

K

key (mtlearn.data.DatasetSpec attribute)

L

list_datasets() (in module mtlearn.data) load_checkpoint() (in module mtlearn.layers) load_stats() (mtlearn.layers.ConnectedFilterPreprocessingLayer method) (mtlearn.layers.ConnectedFilterPreprocessingLayerLegacy method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithCPUTreeTraversal method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithExplicitJacobian method)

local_env_var (mtlearn.data.DatasetSpec attribute)

M

main() (in module mtlearn.data) MAX_TREE (mtlearn.morphology.TreeType attribute) MIN_TREE (mtlearn.morphology.TreeType attribute) module mtlearn.data mtlearn.datasets mtlearn.morphology

mtlearn.data module mtlearn.datasets module mtlearn.morphology module

N

normalize_tos_interpolation() (in module mtlearn.morphology)

normalize_tree_type() (in module mtlearn.morphology)

P

PairedImageDataset (class in mtlearn.datasets) parse_args() (in module mtlearn.data) predict() (mtlearn.layers.ConnectedFilterPreprocessingLayer method) (mtlearn.layers.ConnectedFilterPreprocessingLayerLegacy method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithCPUTreeTraversal method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithExplicitJacobian method)

R

refresh_cached_normalization() (mtlearn.layers.ConnectedFilterPreprocessingLayer method) (mtlearn.layers.ConnectedFilterPreprocessingLayerLegacy method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithCPUTreeTraversal method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithExplicitJacobian method)

replace_directory() (in module mtlearn.data) repo_root() (in module mtlearn.data) require_local_dataset() (in module mtlearn.data)

S

save_checkpoint() (in module mtlearn.layers) save_params() (mtlearn.layers.ConnectedFilterPreprocessingLayer method) (mtlearn.layers.ConnectedFilterPreprocessingLayerLegacy method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithCPUTreeTraversal method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithExplicitJacobian method)

save_stats() (mtlearn.layers.ConnectedFilterPreprocessingLayer method) (mtlearn.layers.ConnectedFilterPreprocessingLayerLegacy method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithCPUTreeTraversal method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithExplicitJacobian method) set_extra_state() (mtlearn.layers.ConnectedFilterPreprocessingLayer method)

T

target (mtlearn.data.DatasetSpec attribute) train_test_split() (mtlearn.datasets.AttributeFilterDataset method) (mtlearn.datasets.PairedImageDataset method)

TREE_OF_SHAPES (mtlearn.morphology.TreeType attribute) TreeType (class in mtlearn.morphology)

U

unfreeze_ds_stats() (mtlearn.layers.ConnectedFilterPreprocessingLayer method) (mtlearn.layers.ConnectedFilterPreprocessingLayerLegacy method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithCPUTreeTraversal method) (mtlearn.layers.ConnectedFilterPreprocessingLayerWithExplicitJacobian method)

url (mtlearn.data.DatasetSpec attribute) url_env_var (mtlearn.data.DatasetSpec attribute)

W

WeightedMorphologicalTree (class in mtlearn.morphology)