SYMBOL INDEX (28 symbols across 9 files)

FILE: examples/mnist/mnist.py
  class CachedMNIST (line 21) | class CachedMNIST(Dataset):
    method __init__ (line 22) | def __init__(self, train, cuda, testing_mode=False):
    method _transformation (line 30) | def _transformation(img):
    method __getitem__ (line 36) | def __getitem__(self, index: int) -> torch.Tensor:
    method __len__ (line 46) | def __len__(self) -> int:
  function main (line 75) | def main(cuda, batch_size, pretrain_epochs, finetune_epochs, testing_mode):

FILE: ptdec/cluster.py
  class ClusterAssignment (line 7) | class ClusterAssignment(nn.Module):
    method __init__ (line 8) | def __init__(
    method forward (line 38) | def forward(self, batch: torch.Tensor) -> torch.Tensor:

FILE: ptdec/dec.py
  class DEC (line 7) | class DEC(nn.Module):
    method __init__ (line 8) | def __init__(
    method forward (line 33) | def forward(self, batch: torch.Tensor) -> torch.Tensor:

FILE: ptdec/model.py
  function train (line 12) | def train(
  function predict (line 176) | def predict(

FILE: ptdec/utils.py
  function cluster_accuracy (line 7) | def cluster_accuracy(y_true, y_predicted, cluster_number: Optional[int] ...
  function target_distribution (line 31) | def target_distribution(batch: torch.Tensor) -> torch.Tensor:

FILE: tests/test_cluster.py
  class TestClusterAssignment (line 7) | class TestClusterAssignment(TestCase):
    method setUpClass (line 9) | def setUpClass(cls):
    method test_forward (line 16) | def test_forward(self):

FILE: tests/test_dec.py
  class TestAutoEncoder (line 9) | class TestAutoEncoder(TestCase):
    method setUpClass (line 11) | def setUpClass(cls):
    method test_dimension (line 15) | def test_dimension(self):

FILE: tests/test_model.py
  function test_train_with_prediction (line 7) | def test_train_with_prediction():
  function test_predict (line 26) | def test_predict():

FILE: tests/test_utils.py
  class TestClusterAccuracy (line 8) | class TestClusterAccuracy(TestCase):
    method test_basic (line 9) | def test_basic(self):
  class TestTargetDistribution (line 33) | class TestTargetDistribution(TestCase):
    method test_basic (line 34) | def test_basic(self):