Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/NetManAIOps/donut
/ types & classes
Types & classes
18 in github.com/NetManAIOps/donut
⨍
Functions
68
◇
Types & classes
18
↓ 18 callers
Class
BatchSlidingWindow
Class for obtaining mini-batch iterators of sliding windows. Each mini-batch will have `batch_size` windows. If the final batch contain
donut/utils.py:31
↓ 12 callers
Class
Donut
Class for constructing Donut model. This class provides :meth:`get_training_loss` for deriving the training loss :class:`tf.Tensor`, and
donut/model.py:31
↓ 6 callers
Class
DonutTrainer
Donut trainer. Args: model (Donut): The :class:`Donut` model instance. model_vs (str or tf.VariableScope): If specified, wil
donut/training.py:18
↓ 4 callers
Class
DataAugmentation
Base class for data augmentation in training. Args: mean (float): Mean of the training data. std (float): Standard deviation
donut/augmentation.py:8
↓ 3 callers
Class
DonutPredictor
Donut predictor. Args: model (Donut): The :class:`Donut` model instance. n_z (int or None): Number of `z` samples to take fo
donut/prediction.py:13
↓ 3 callers
Class
MissingDataInjection
Data augmentation by injecting missing points into training data. Args: mean (float): Mean of the training data. std (float)
donut/augmentation.py:70
↓ 2 callers
Class
Capture
tests/test_model.py:73
Class
AugmentationTestCase
tests/test_augmentation.py:11
Class
BatchSlidingWindowTestCase
tests/test_utils.py:34
Class
CompleteTimestampTestCase
tests/test_preprocessing.py:9
Class
DonutPredictorTestCase
Tests for :class:`DonutPredictor`. It is very difficult to check the outputs of a :class:`DonutPredictor`, since it involves random samp
tests/test_prediction.py:9
Class
DonutTrainerTestCase
Tests for :class:`DonutTrainer`. It is very difficult to test a :class:`DonutTrainer`. We just check whether or not the :class:`DonutTra
tests/test_training.py:9
Class
IterativeMaskedReconstructTestCase
tests/test_reconstruction.py:79
Class
MaskedReconstructTestCase
tests/test_reconstruction.py:9
Class
MiniBatchSlicesIteratorTestCase
tests/test_utils.py:9
Class
MissingDataInjectionTestCase
tests/test_augmentation.py:48
Class
ModelTestCase
tests/test_model.py:12
Class
StandardizeKPITestCase
tests/test_preprocessing.py:72