SYMBOL INDEX (78 symbols across 7 files)

FILE: kaggle_dsb18/kaggle_dsb18_preprocessing.py
  function chk_mkdir (line 20) | def chk_mkdir(*paths: Container) -> None:
  function merge_masks (line 32) | def merge_masks(masks_folder):

FILE: unet/blocks.py
  class SoftDiceLoss (line 5) | class SoftDiceLoss(_Loss):
    method __init__ (line 6) | def __init__(self, size_average=None, reduce=None, reduction='mean'):
    method forward (line 9) | def forward(self, y_pred, y_gt):
  class First2D (line 15) | class First2D(nn.Module):
    method __init__ (line 16) | def __init__(self, in_channels, middle_channels, out_channels, dropout...
    method forward (line 34) | def forward(self, x):
  class Encoder2D (line 38) | class Encoder2D(nn.Module):
    method __init__ (line 39) | def __init__(
    method forward (line 61) | def forward(self, x):
  class Center2D (line 65) | class Center2D(nn.Module):
    method __init__ (line 66) | def __init__(self, in_channels, middle_channels, out_channels, deconv_...
    method forward (line 86) | def forward(self, x):
  class Decoder2D (line 90) | class Decoder2D(nn.Module):
    method __init__ (line 91) | def __init__(self, in_channels, middle_channels, out_channels, deconv_...
    method forward (line 110) | def forward(self, x):
  class Last2D (line 114) | class Last2D(nn.Module):
    method __init__ (line 115) | def __init__(self, in_channels, middle_channels, out_channels, softmax...
    method forward (line 131) | def forward(self, x):
  class First3D (line 135) | class First3D(nn.Module):
    method __init__ (line 136) | def __init__(self, in_channels, middle_channels, out_channels, dropout...
    method forward (line 154) | def forward(self, x):
  class Encoder3D (line 158) | class Encoder3D(nn.Module):
    method __init__ (line 159) | def __init__(
    method forward (line 181) | def forward(self, x):
  class Center3D (line 185) | class Center3D(nn.Module):
    method __init__ (line 186) | def __init__(self, in_channels, middle_channels, out_channels, deconv_...
    method forward (line 206) | def forward(self, x):
  class Decoder3D (line 210) | class Decoder3D(nn.Module):
    method __init__ (line 211) | def __init__(self, in_channels, middle_channels, out_channels, deconv_...
    method forward (line 230) | def forward(self, x):
  class Last3D (line 234) | class Last3D(nn.Module):
    method __init__ (line 235) | def __init__(self, in_channels, middle_channels, out_channels, softmax...
    method forward (line 251) | def forward(self, x):

FILE: unet/dataset.py
  function to_long_tensor (line 14) | def to_long_tensor(pic):
  function correct_dims (line 21) | def correct_dims(*images):
  class JointTransform2D (line 35) | class JointTransform2D:
    method __init__ (line 52) | def __init__(self, crop=(256, 256), p_flip=0.5, color_jitter_params=(0...
    method __call__ (line 62) | def __call__(self, image, mask):
  class ImageToImage2D (line 93) | class ImageToImage2D(Dataset):
    method __init__ (line 120) | def __init__(self, dataset_path: str, joint_transform: Callable = None...
    method __len__ (line 133) | def __len__(self):
    method __getitem__ (line 136) | def __getitem__(self, idx):
  class Image2D (line 156) | class Image2D(Dataset):
    method __init__ (line 179) | def __init__(self, dataset_path: str, transform: Callable = None):
    method __len__ (line 189) | def __len__(self):
    method __getitem__ (line 192) | def __getitem__(self, idx):

FILE: unet/metrics.py
  class LogNLLLoss (line 9) | class LogNLLLoss(_WeightedLoss):
    method __init__ (line 12) | def __init__(self, weight=None, size_average=None, reduce=None, reduct...
    method forward (line 17) | def forward(self, y_input, y_target):
  function classwise_iou (line 23) | def classwise_iou(output, gt):
  function classwise_f1 (line 38) | def classwise_f1(output, gt):
  function make_weighted_metric (line 60) | def make_weighted_metric(classwise_metric):

FILE: unet/model.py
  class Model (line 18) | class Model:
    method __init__ (line 33) | def __init__(self, net: nn.Module, loss, optimizer, checkpoint_folder:...
    method fit_epoch (line 73) | def fit_epoch(self, dataset, n_batch=1, shuffle=False):
    method val_epoch (line 112) | def val_epoch(self, dataset, n_batch=1, metric_list=MetricList({})):
    method fit_dataset (line 148) | def fit_dataset(self, dataset: ImageToImage2D, n_epochs: int, n_batch:...
    method predict_dataset (line 229) | def predict_dataset(self, dataset, export_path):

FILE: unet/unet.py
  class UNet2D (line 8) | class UNet2D(nn.Module):
    method __init__ (line 9) | def __init__(self, in_channels, out_channels, conv_depths=(64, 128, 25...
    method forward (line 31) | def forward(self, x, return_all=False):
  class UNet3D (line 51) | class UNet3D(nn.Module):
    method __init__ (line 52) | def __init__(self, in_channels, out_channels, conv_depths=(64, 128, 25...
    method forward (line 74) | def forward(self, x, return_all=False):
  function pad_to_shape (line 94) | def pad_to_shape(this, shp):

FILE: unet/utils.py
  function chk_mkdir (line 10) | def chk_mkdir(*paths: Container) -> None:
  class Logger (line 22) | class Logger:
    method __init__ (line 23) | def __init__(self, verbose=False):
    method log (line 27) | def log(self, logs):
    method get_logs (line 34) | def get_logs(self):
    method to_csv (line 37) | def to_csv(self, path):
  class MetricList (line 41) | class MetricList:
    method __init__ (line 42) | def __init__(self, metrics):
    method __call__ (line 47) | def __call__(self, y_out, y_batch):
    method reset (line 51) | def reset(self):
    method get_results (line 54) | def get_results(self, normalize=False):