timesead.models.baselines.threshold_ads
Classes
Base class for all neural network modules. |
|
Base class for all neural network modules. |
|
Base class for all neural network modules. |
Module Contents
- class timesead.models.baselines.threshold_ads.Base_ThresholdAD(first_diffs: bool, cum_method: str, feature_index: int | None, input_shape: str = 'btf')
Bases:
timesead.models.common.AnomalyDetectorBase class for all neural network modules.
Your models should also subclass this class.
Modules can also contain other Modules, allowing them to be nested in a tree structure. You can assign the submodules as regular attributes:
import torch.nn as nn import torch.nn.functional as F class Model(nn.Module): def __init__(self) -> None: super().__init__() self.conv1 = nn.Conv2d(1, 20, 5) self.conv2 = nn.Conv2d(20, 20, 5) def forward(self, x): x = F.relu(self.conv1(x)) return F.relu(self.conv2(x))
Submodules assigned in this way will be registered, and will also have their parameters converted when you call
to(), etc.Note
As per the example above, an
__init__()call to the parent class must be made before assignment on the child.- Variables:
training (bool) – Boolean represents whether this module is in training or evaluation mode.
- Parameters:
- A Basis Threshold Anomaly Detector
A simlpe anomaly detector that is equals zero for all data within the given Threshold and else equals the distance to the given lower/upper thr of training value.
- Parameters:
first_diffs[bool] – Flag, if instead of raw values first difference should be used.
cum_method[str] – One of {‘mean’, ‘max’}. Accumulation method over feature dimension. Note that when feature_index is not None. Then accumulation method will be ignored.
feature_index[optional[int]] – Take scores for specific feature. If none, above accumulation rule will be used.
first_diffs (bool)
cum_method (str)
feature_index (Optional[int])
input_shape (str)
- first_diffs
- cum_method
- feature
- input_shape = 'btf'
- lower_thresh = None
- upper_thresh = None
- compute_online_anomaly_score(inputs: Tuple[torch.Tensor, Ellipsis]) torch.Tensor
Compute the online anomaly score for a batch of inputs. The output tensor must have the same shape as the output of format_targets when called with the corresponding targets for this batch. This method expects a window (or a batch of windows) as its input and should return a score for the last point in the window.
- Parameters:
inputs (Tuple[torch.Tensor, Ellipsis]) – tuple of input tensors
- Returns:
Tensor of shape (B,) that contains the anomaly scores for this batch
- Return type:
- abstract compute_offline_anomaly_score(inputs: Tuple[torch.Tensor, Ellipsis]) torch.Tensor
Compute the offline anomaly score for a batch of inputs. The output tensor must have the same shape as the output of format_targets when called with the corresponding targets for this batch. This method expects a window (or a batch of windows) as its input and should return a score for the last point in the window.
- Parameters:
inputs (Tuple[torch.Tensor, Ellipsis]) – tuple of input tensors
- Returns:
Tensor of shape (N,) that contains the anomaly scores for this batch
- Return type:
- format_online_targets(targets: Tuple[torch.Tensor, Ellipsis]) torch.Tensor
Format the labels for a batch of targets. The output tensor must have the same shape as the output of compute_online_anomaly_score when called with the corresponding inputs for this batch.
- Parameters:
targets (Tuple[torch.Tensor, Ellipsis]) – tuple of target tensors
- Returns:
Tensor of shape (B,) that contains the ground truth labels for this batch
- Return type:
- class timesead.models.baselines.threshold_ads.OOSAnomalyDetector(first_diffs: bool = False, cum_method: str = 'mean', feature_index: int | None = None, *args, **kwargs)
Bases:
Base_ThresholdADBase class for all neural network modules.
Your models should also subclass this class.
Modules can also contain other Modules, allowing them to be nested in a tree structure. You can assign the submodules as regular attributes:
import torch.nn as nn import torch.nn.functional as F class Model(nn.Module): def __init__(self) -> None: super().__init__() self.conv1 = nn.Conv2d(1, 20, 5) self.conv2 = nn.Conv2d(20, 20, 5) def forward(self, x): x = F.relu(self.conv1(x)) return F.relu(self.conv2(x))
Submodules assigned in this way will be registered, and will also have their parameters converted when you call
to(), etc.Note
As per the example above, an
__init__()call to the parent class must be made before assignment on the child.- Variables:
training (bool) – Boolean represents whether this module is in training or evaluation mode.
- Parameters:
- Out of Support Anomaly Detector
A simlpe anomaly detector that is equals zero for all data within the data support of the training data and else equals the distance to the min/max of training value.
- Parameters:
first_diffs[bool] – Flag, if instead of raw values first difference should be used. Default is False.
cum_method[str] – One of {‘mean’, ‘max’}. Accumulation method over feature dimension. Note that when feature_index is not None. Then accumulation method will be ignored. Default is ‘mean’.
feature_index[optional[int]] – Take scores for specific feature. If none, above accumulation rule will be used. Default is None.
first_diffs (bool)
cum_method (str)
feature_index (Optional[int])
- fit(dataset: torch.utils.data.DataLoader) None
Fit this anomaly detector on a dataset. Note that we assume only normal data here.
- Parameters:
dataset (torch.utils.data.DataLoader) – A dataset
- Return type:
None
- class timesead.models.baselines.threshold_ads.IQRAnomalyDetector(std_factor: float = 2.58, first_diffs: bool = False, cum_method: str = 'mean', feature_index: int | None = None, *args, **kwargs)
Bases:
Base_ThresholdADBase class for all neural network modules.
Your models should also subclass this class.
Modules can also contain other Modules, allowing them to be nested in a tree structure. You can assign the submodules as regular attributes:
import torch.nn as nn import torch.nn.functional as F class Model(nn.Module): def __init__(self) -> None: super().__init__() self.conv1 = nn.Conv2d(1, 20, 5) self.conv2 = nn.Conv2d(20, 20, 5) def forward(self, x): x = F.relu(self.conv1(x)) return F.relu(self.conv2(x))
Submodules assigned in this way will be registered, and will also have their parameters converted when you call
to(), etc.Note
As per the example above, an
__init__()call to the parent class must be made before assignment on the child.- Variables:
training (bool) – Boolean represents whether this module is in training or evaluation mode.
- Parameters:
- Interquantile Range Anomaly Detector
A simlpe anomaly detector that is equals zero for all data within some interquantile range of the normal training data and else equals the distance to nearest quantile border.
- Parameters:
std_factor[float] – float, that gives the width of the IQR. Default is 2.58 (=99.5% normal quantile).
first_diffs[bool] – Flag, if instead of raw values first difference should be used. Default is False.
cum_method[str] – One of {‘mean’, ‘max’}. Accumulation method over feature dimension. Note that when feature_index is not None. Then accumulation method will be ignored. Default is ‘mean’.
feature_index[optional[int]] – Take scores for specific feature. If none, above accumulation rule will be used. Default is None.
std_factor (float)
first_diffs (bool)
cum_method (str)
feature_index (Optional[int])
- std_factor = 2.58
- fit(dataset: torch.utils.data.DataLoader) None
Fit this anomaly detector on a dataset. Note that we assume only normal data here.
- Parameters:
dataset (torch.utils.data.DataLoader) – A dataset
- Return type:
None