id: 1
unit: def train()
file: code/train.py
start line: 0
end line: 0
size: 71 LOC
McCabe index: 13
number of parameters: 1

id: 2
unit: def test()
file: code/train.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 2
number of parameters: 3

id: 3
unit: def __getitem__()
file: code/data_prep.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 1
number of parameters: 2

id: 4
unit: def input_fn()
file: code/inference.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 2
number of parameters: 2

id: 5
unit: def _get_train_data_loader()
file: code/train.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 1
number of parameters: 2

id: 6
unit: def predict_fn()
file: code/inference.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 2

id: 7
unit: def _get_test_data_loader()
file: code/train.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 2

id: 8
unit: def model_fn()
file: code/inference.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 1

id: 9
unit: def __init__()
file: code/model_def.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

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

id: 11
unit: def freeze()
file: code/train.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 2

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

id: 13
unit: def save_model()
file: code/train.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 14
unit: def __len__()
file: code/data_prep.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1