src.ml package

Submodules

src.ml.pytorch_example module

From: https://pytorch.org/tutorials/beginner/text_sentiment_ngrams_tutorial.html

src.ml.pytorch_example.BATCH_SIZE = 16

class src.ml.pytorch_example.TextSentiment(vocab_size, embed_dim, num_class)

Bases: torch.nn.modules.module.Module

forward(text, offsets)

init_weights()

src.ml.pytorch_example.device = device(type='cpu')

src.ml.pytorch_example.generate_batch(batch)

src.ml.pytorch_example.main()

src.ml.pytorch_example.test(model, criterion, data_)

src.ml.pytorch_example.train_func(model, optimizer, criterion, scheduler, sub_train_)