id: 1
unit: def _create_base_tf_dataset()
file: projects/home/recap/data/dataset.py
start line: 0
end line: 0
size: 75 LOC
McCabe index: 15
number of parameters: 2

id: 2
unit: def progress()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 59 LOC
McCabe index: 20
number of parameters: 2

id: 3
unit: def run()
file: projects/home/recap/main.py
start line: 0
end line: 0
size: 53 LOC
McCabe index: 3
number of parameters: 2

id: 4
unit: def _create_tf_dataset()
file: projects/home/recap/data/dataset.py
start line: 0
end line: 0
size: 48 LOC
McCabe index: 10
number of parameters: 1

id: 5
unit: def compute()
file: metrics/auroc.py
start line: 0
end line: 0
size: 47 LOC
McCabe index: 5
number of parameters: 1

id: 6
unit: def to_batch()
file: projects/home/recap/data/dataset.py
start line: 0
end line: 0
size: 38 LOC
McCabe index: 13
number of parameters: 2

id: 7
unit: def forward()
file: projects/twhin/models/models.py
start line: 0
end line: 0
size: 35 LOC
McCabe index: 7
number of parameters: 2

id: 8
unit: def forward()
file: projects/home/recap/model/model_and_loss.py
start line: 0
end line: 0
size: 32 LOC
McCabe index: 4
number of parameters: 2

id: 9
unit: def maybe_start_dataset_service()
file: reader/dds.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 5
number of parameters: 0

id: 10
unit: def build_preprocess()
file: projects/home/recap/data/preprocessors.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 10
number of parameters: 2

id: 11
unit: def compute_lr()
file: optimizers/optimizer.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 7
number of parameters: 2

id: 12
unit: def rank_specific()
file: ml_logging/torch_logging.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 7
number of parameters: 1

id: 13
unit: def build_optimizer()
file: projects/twhin/optimizer.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 2
number of parameters: 2

id: 14
unit: def progress()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 5
number of parameters: 2

id: 15
unit: def __call__()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 7
number of parameters: 3

id: 16
unit: def call()
file: projects/home/recap/data/preprocessors.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 9
number of parameters: 4

id: 17
unit: def main()
file: machines/get_env.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 14
number of parameters: 1

id: 18
unit: def __init__()
file: projects/home/recap/model/mask_net.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 5
number of parameters: 3

id: 19
unit: def __init__()
file: projects/home/recap/model/mlp.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 5
number of parameters: 3

id: 20
unit: def checkpoints_iterator()
file: common/checkpointing/snapshot.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 7
number of parameters: 3

id: 21
unit: def get_explicit_datetime_inputs_files()
file: projects/home/recap/data/dataset.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 7
number of parameters: 1

id: 22
unit: def speed_check()
file: reader/utils.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 6
number of parameters: 4

id: 23
unit: def create_default_pa_to_batch()
file: reader/utils.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 3
number of parameters: 1

id: 24
unit: def train()
file: core/custom_training_loop.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 1
number of parameters: 0

id: 25
unit: def loss_fn()
file: core/losses.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 3
number of parameters: 3

id: 26
unit: def forward()
file: projects/twhin/models/models.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 1
number of parameters: 2

id: 27
unit: def forward()
file: projects/home/recap/model/mask_net.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 6
number of parameters: 2

id: 28
unit: def to_batches()
file: projects/twhin/data/edges.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 2
number of parameters: 1

id: 29
unit: def __init__()
file: projects/home/recap/data/preprocessors.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 3
number of parameters: 2

id: 30
unit: def call()
file: projects/home/recap/data/preprocessors.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 5
number of parameters: 4

id: 31
unit: def roundrobin()
file: reader/utils.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 6
number of parameters: 1

id: 32
unit: def get_seg_dense_parse_fn()
file: projects/home/recap/data/tfe_parsing.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 1
number of parameters: 1

id: 33
unit: def apply_optimizers()
file: projects/twhin/models/models.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 4
number of parameters: 2

id: 34
unit: def get_global_loss_detached()
file: core/losses.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 4
number of parameters: 2

id: 35
unit: def get_checkpoints()
file: common/checkpointing/snapshot.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 4
number of parameters: 1

id: 36
unit: def only_evaluate()
file: core/custom_training_loop.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 0

id: 37
unit: def _connect()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 2

id: 38
unit: def __init__()
file: projects/twhin/models/models.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 3

id: 39
unit: def setup_and_get_device()
file: common/device.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 4
number of parameters: 1

id: 40
unit: def mock_pg()
file: common/testing_utils.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 0

id: 41
unit: def _validate_mtl()
file: projects/home/recap/model/config.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 6
number of parameters: 2

id: 42
unit: def generate_data()
file: projects/home/recap/data/generate_random_data.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 2

id: 43
unit: def pa_to_batch()
file: projects/twhin/data/edges.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 2

id: 44
unit: def get_readers()
file: reader/dataset.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 1

id: 45
unit: def prepend_transform()
file: core/metric_mixin.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 2

id: 46
unit: def forward()
file: projects/home/recap/model/entrypoint.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 47
unit: def __init__()
file: projects/home/recap/model/feature_transform.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 4

id: 48
unit: def call()
file: projects/home/recap/data/preprocessors.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 4

id: 49
unit: def call()
file: projects/home/recap/data/preprocessors.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 4

id: 50
unit: def main()
file: projects/twhin/run.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 1

id: 51
unit: def create_dataset()
file: projects/twhin/data/data.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 2

id: 52
unit: def forward()
file: model.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 2

id: 53
unit: def restore()
file: common/checkpointing/snapshot.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 2

id: 54
unit: def do_get()
file: reader/dataset.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 3

id: 55
unit: def __init__()
file: reader/dataset.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 3

id: 56
unit: def maybe_distribute_dataset()
file: reader/dds.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 1

id: 57
unit: def _add_weights()
file: projects/home/recap/data/dataset.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 3

id: 58
unit: def get_datetimes()
file: projects/home/recap/data/dataset.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 1

id: 59
unit: def _add_weights_based_on_sampling_rates()
file: projects/home/recap/data/dataset.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 3

id: 60
unit: def save()
file: common/checkpointing/snapshot.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 2
number of parameters: 2

id: 61
unit: def get_dds_worker_address()
file: machines/environment.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 0

id: 62
unit: def to_batches()
file: reader/dataset.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 1

id: 63
unit: def _get_step_fn()
file: core/custom_training_loop.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 3

id: 64
unit: def _wait_for_batch()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 2

id: 65
unit: def transform()
file: core/metrics.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 3

id: 66
unit: def _build_single_task_model()
file: projects/home/recap/model/entrypoint.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 2

id: 67
unit: def _compute_jagged_tensor_from_tensor()
file: projects/home/recap/data/util.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 1

id: 68
unit: def _init_tensor_spec()
file: projects/home/recap/data/dataset.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 1

id: 69
unit: def __init__()
file: projects/twhin/data/edges.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 70
unit: def get_optimizer_algorithm_config()
file: optimizers/config.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 1

id: 71
unit: def get_dds_dispatcher_address()
file: machines/environment.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 0

id: 72
unit: def get_flight_server_addresses()
file: machines/environment.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 0

id: 73
unit: def stable_mean_dist_reduce_fn()
file: metrics/aggregation.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 1

id: 74
unit: def __iter__()
file: tools/pq.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 1

id: 75
unit: def bytes_per_row()
file: tools/pq.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 1

id: 76
unit: def _get_unsharded_module_names()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 1

id: 77
unit: def __init__()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 78
unit: def train()
file: core/debug_training_loop.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 6

id: 79
unit: def __init__()
file: projects/home/recap/model/feature_transform.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 3

id: 80
unit: def _dense_tf_to_torch()
file: projects/home/recap/data/util.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 2

id: 81
unit: def __init__()
file: projects/home/recap/data/dataset.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 82
unit: def batch_size()
file: common/batch.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 1

id: 83
unit: def maybe_run_training()
file: common/run_training.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 84
unit: def main()
file: machines/list_ops.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 1

id: 85
unit: def get_dds()
file: machines/environment.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 0

id: 86
unit: def _create_dataset()
file: reader/dataset.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 1

id: 87
unit: def __init__()
file: metrics/aggregation.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 2

id: 88
unit: def update()
file: metrics/aggregation.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 4

id: 89
unit: def compute()
file: metrics/rce.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 1

id: 90
unit: def _compute_helper()
file: metrics/auroc.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 91
unit: def __init__()
file: metrics/auroc.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 4

id: 92
unit: def _head()
file: tools/pq.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 93
unit: def _run_evaluation()
file: core/custom_training_loop.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 94
unit: def __init__()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 95
unit: def _field_data_map()
file: core/config/base_config.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 2

id: 96
unit: def load_config_from_yaml()
file: core/config/config_load.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 2

id: 97
unit: def _maybe_warn()
file: core/losses.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 98
unit: def __init__()
file: projects/home/recap/optimizer/optimizer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 99
unit: def get_lr()
file: projects/home/recap/optimizer/optimizer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 100
unit: def _get_closed_form_lr()
file: projects/home/recap/optimizer/optimizer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 101
unit: def create_ranking_model()
file: projects/home/recap/model/entrypoint.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 0

id: 102
unit: def forward()
file: projects/home/recap/model/mlp.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 2

id: 103
unit: def __init__()
file: projects/home/recap/data/preprocessors.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 2

id: 104
unit: def _serialize_example()
file: projects/home/recap/data/generate_random_data.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 2

id: 105
unit: def embedding_dims_match()
file: projects/twhin/models/config.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 2

id: 106
unit: def get_lr()
file: optimizers/optimizer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 107
unit: def get_optimizer_class()
file: optimizers/optimizer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 1

id: 108
unit: def maybe_setup_tensorflow()
file: common/device.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 0

id: 109
unit: def infer_fs()
file: common/filesystem/util.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 110
unit: def get_num_readers()
file: machines/environment.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 4
number of parameters: 0

id: 111
unit: def _main()
file: machines/is_venv.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 0

id: 112
unit: def setup_absl_logging()
file: ml_logging/absl_logging.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 113
unit: def distribute_from_dataset_id()
file: reader/dds.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 114
unit: def forward()
file: metrics/rce.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 3

id: 115
unit: def _connect()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 116
unit: def transform()
file: core/metrics.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 117
unit: def transform()
file: core/metrics.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 118
unit: def transform()
file: core/metrics.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 119
unit: def transform()
file: core/metrics.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 3

id: 120
unit: def transform()
file: core/metrics.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 3

id: 121
unit: def transform()
file: core/metrics.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 3

id: 122
unit: def _one_of_check()
file: core/config/base_config.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 2

id: 123
unit: def _at_most_one_of_check()
file: core/config/base_config.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 2

id: 124
unit: def build_multi_task_loss()
file: core/losses.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 125
unit: def update()
file: core/metric_mixin.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 5
number of parameters: 3

id: 126
unit: def __init__()
file: projects/home/recap/model/entrypoint.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 127
unit: def _validate_evaluation_tasks()
file: projects/home/recap/data/config.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 2

id: 128
unit: def __post_init__()
file: projects/home/recap/data/dataset.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 5
number of parameters: 1

id: 129
unit: def _chain()
file: projects/home/recap/data/dataset.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 3

id: 130
unit: def __init__()
file: projects/twhin/models/models.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 131
unit: def _lr_from_config()
file: projects/twhin/optimizer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 132
unit: def __init__()
file: optimizers/optimizer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 133
unit: def log_sharded_tensor_content()
file: model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 3

id: 134
unit: def as_dict()
file: common/batch.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 1

id: 135
unit: def from_schema()
file: common/batch.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 136
unit: def from_fields()
file: common/batch.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 137
unit: def get_task_index()
file: machines/environment.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 0

id: 138
unit: def _validate_columns()
file: reader/dataset.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 139
unit: def dataloader()
file: reader/dataset.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 2

id: 140
unit: def update_mean()
file: metrics/aggregation.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 141
unit: def _binary_cross_entropy_with_clipping()
file: metrics/rce.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 142
unit: def update()
file: metrics/rce.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 143
unit: def update()
file: metrics/auroc.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 144
unit: def log_eval_results()
file: core/custom_training_loop.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 145
unit: def _to_device()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 3

id: 146
unit: def __init__()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 147
unit: def forward()
file: projects/home/recap/model/mask_net.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 148
unit: def __init__()
file: projects/home/recap/model/model_and_loss.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 149
unit: def __init__()
file: projects/home/recap/model/feature_transform.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 150
unit: def _generate_data_main()
file: projects/home/recap/data/generate_random_data.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 151
unit: def valid_node_types()
file: projects/twhin/models/config.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 4

id: 152
unit: def to()
file: common/batch.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 153
unit: def pin_memory()
file: common/batch.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 154
unit: def get_torch_snapshot()
file: common/checkpointing/snapshot.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 155
unit: def load_snapshot_to_weight()
file: common/checkpointing/snapshot.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 156
unit: def wait_for_evaluators()
file: common/checkpointing/snapshot.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 157
unit: def has_readers()
file: machines/environment.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 0

id: 158
unit: def __init__()
file: reader/dataset.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 159
unit: def reset()
file: metrics/rce.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 160
unit: def _create_dataset()
file: tools/pq.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 161
unit: def distinct()
file: tools/pq.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 162
unit: def is_leaf_module()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 3

id: 163
unit: def _start_data_dist()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 164
unit: def _get_node_args_helper()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 165
unit: def _get_unsharded_module_names_helper()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 166
unit: def _rewrite_model()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 167
unit: def _sync_pipeline()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 168
unit: def __init__()
file: core/metric_mixin.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 169
unit: def build_optimizer()
file: projects/home/recap/optimizer/optimizer.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 170
unit: def __init__()
file: projects/home/recap/model/numeric_calibration.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 171
unit: def _init_weights()
file: projects/home/recap/model/mask_net.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 172
unit: def __init__()
file: projects/home/recap/model/feature_transform.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 173
unit: def _init_weights()
file: projects/home/recap/model/mlp.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 174
unit: def __init__()
file: projects/home/recap/data/preprocessors.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 175
unit: def call()
file: projects/home/recap/data/preprocessors.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 4

id: 176
unit: def parse_tf_example()
file: projects/home/recap/data/tfe_parsing.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 177
unit: def __init__()
file: model.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 178
unit: def __repr__()
file: common/batch.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 1

id: 179
unit: def log_ebc_norms()
file: common/log_weights.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 180
unit: def setup_configuration()
file: common/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 181
unit: def _substitute()
file: common/utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 182
unit: def __init__()
file: common/checkpointing/snapshot.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 4

id: 183
unit: def get_checkpoint()
file: common/checkpointing/snapshot.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 184
unit: def is_distributed_worker()
file: common/run_training.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 0

id: 185
unit: def get_task_type()
file: machines/environment.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 0

id: 186
unit: def get_reader_port()
file: machines/environment.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 0

id: 187
unit: def serve()
file: reader/dataset.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 188
unit: def reset()
file: metrics/rce.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 189
unit: def __init__()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

id: 190
unit: def probs_and_labels()
file: core/metrics.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 191
unit: def transform()
file: core/metrics.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 192
unit: def transform()
file: core/metrics.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 193
unit: def transform()
file: core/metrics.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 194
unit: def __init__()
file: core/metrics.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 195
unit: def __init__()
file: core/metrics.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 196
unit: def build_loss()
file: core/losses.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 197
unit: def __init__()
file: core/metric_mixin.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 198
unit: def __init__()
file: projects/home/recap/model/feature_transform.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 199
unit: def jagged_tensor_from_tensor()
file: projects/home/recap/data/util.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 200
unit: def _gen()
file: projects/home/recap/data/dataset.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 201
unit: def __init__()
file: projects/home/recap/data/preprocessors.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 202
unit: def __init__()
file: projects/home/recap/data/preprocessors.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 203
unit: def create_tf_example_schema()
file: projects/home/recap/data/tfe_parsing.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 204
unit: def mask_mantissa()
file: projects/home/recap/data/tfe_parsing.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 205
unit: def run()
file: projects/twhin/run.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 206
unit: def maybe_shard_model()
file: model.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 207
unit: def record_stream()
file: common/batch.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

id: 208
unit: def weights_to_log()
file: common/log_weights.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 209
unit: def _read_file()
file: common/utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 210
unit: def __init__()
file: common/modules/embedding/embedding.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 211
unit: def forward()
file: common/modules/embedding/embedding.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 212
unit: def pa_to_batch()
file: reader/dataset.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 213
unit: def pa_to_torch()
file: reader/utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 214
unit: def register_dataset()
file: reader/dds.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 215
unit: def compute()
file: metrics/aggregation.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 216
unit: def _smooth()
file: metrics/rce.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 217
unit: def __init__()
file: metrics/rce.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 218
unit: def update()
file: metrics/rce.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 219
unit: def __init__()
file: metrics/rce.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 220
unit: def __init__()
file: tools/pq.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 221
unit: def schema()
file: tools/pq.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 222
unit: def head()
file: tools/pq.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 223
unit: def get_new_iterator()
file: core/custom_training_loop.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 224
unit: def progress()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 225
unit: def name()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 226
unit: def args()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 227
unit: def _get_node_args()
file: core/train_pipeline.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 228
unit: def __init__()
file: core/metrics.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 229
unit: def __init__()
file: core/metrics.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 230
unit: def pretty_print()
file: core/config/base_config.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 231
unit: def loss_fn()
file: core/losses.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 232
unit: def transform()
file: core/metric_mixin.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 233
unit: def maybe_apply_stratification()
file: core/metric_mixin.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 234
unit: def forward()
file: projects/home/recap/model/numeric_calibration.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 235
unit: def __init__()
file: projects/home/recap/model/mask_net.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 236
unit: def sanitize()
file: projects/home/recap/model/entrypoint.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 237
unit: def unsanitize()
file: projects/home/recap/model/entrypoint.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 238
unit: def log_transform()
file: projects/home/recap/model/feature_transform.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 239
unit: def forward()
file: projects/home/recap/model/feature_transform.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 240
unit: def forward()
file: projects/home/recap/model/feature_transform.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 241
unit: def __init__()
file: projects/home/recap/model/feature_transform.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 242
unit: def forward()
file: projects/home/recap/model/feature_transform.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 243
unit: def forward()
file: projects/home/recap/model/feature_transform.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 244
unit: def forward()
file: projects/home/recap/model/feature_transform.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 245
unit: def forward()
file: projects/home/recap/model/feature_transform.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 246
unit: def build_features_preprocessor()
file: projects/home/recap/model/feature_transform.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 247
unit: def shared_size()
file: projects/home/recap/model/mlp.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 248
unit: def out_features()
file: projects/home/recap/model/mlp.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 249
unit: def keyed_tensor_from_tensors_dict()
file: projects/home/recap/data/util.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 250
unit: def keyed_jagged_tensor_from_tensors_dict()
file: projects/home/recap/data/util.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 251
unit: def _tf_to_numpy()
file: projects/home/recap/data/util.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 252
unit: def sparse_or_dense_tf_to_torch()
file: projects/home/recap/data/util.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 253
unit: def _map_output_for_inference()
file: projects/home/recap/data/dataset.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 254
unit: def _map_output_for_train_eval()
file: projects/home/recap/data/dataset.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 255
unit: def to_dataloader()
file: projects/home/recap/data/dataset.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 256
unit: def __iter__()
file: projects/home/recap/data/dataset.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 257
unit: def _generate_random_example()
file: projects/home/recap/data/generate_random_data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 258
unit: def _float_feature()
file: projects/home/recap/data/generate_random_data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 259
unit: def _int64_feature()
file: projects/home/recap/data/generate_random_data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 260
unit: def make_mantissa_mask()
file: projects/home/recap/data/tfe_parsing.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 261
unit: def create_metrics()
file: projects/twhin/metrics.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 262
unit: def _to_kjt()
file: projects/twhin/data/edges.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 263
unit: def _get_closed_form_lr()
file: optimizers/optimizer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 264
unit: def build_optimizer()
file: optimizers/optimizer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 265
unit: def as_dict()
file: common/batch.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 266
unit: def feature_names()
file: common/batch.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 267
unit: def as_dict()
file: common/batch.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 268
unit: def is_local_fs()
file: common/filesystem/util.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 269
unit: def is_gcs_fs()
file: common/filesystem/util.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 270
unit: def step()
file: common/checkpointing/snapshot.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 271
unit: def step()
file: common/checkpointing/snapshot.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 272
unit: def walltime()
file: common/checkpointing/snapshot.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 273
unit: def walltime()
file: common/checkpointing/snapshot.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 274
unit: def _eval_subdir()
file: common/checkpointing/snapshot.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 275
unit: def _eval_done_path()
file: common/checkpointing/snapshot.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 276
unit: def is_done_eval()
file: common/checkpointing/snapshot.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 277
unit: def mark_done_eval()
file: common/checkpointing/snapshot.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 278
unit: def step_from_checkpoint()
file: common/checkpointing/snapshot.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 279
unit: def on_kf()
file: machines/environment.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 280
unit: def is_chief()
file: machines/environment.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 281
unit: def is_reader()
file: machines/environment.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 282
unit: def is_dispatcher()
file: machines/environment.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 283
unit: def get_dds_journaling_dir()
file: machines/environment.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 284
unit: def is_venv()
file: machines/is_venv.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0