commit to merge

This commit is contained in:
xuhongzuo 2022-12-10 15:26:19 +08:00
parent 1af6bcc48d
commit efc80c1e97
3 changed files with 3 additions and 2 deletions

View File

@ -7,6 +7,7 @@ in the skeleton of base deep ad models:
models models
- add AE as a baseline - add AE as a baseline
- add GAN-based method as a baseline - add GAN-based method as a baseline
- add one-class-based methods, e.g., DROCC, HRN, ..
model selection model selection
- add the pretrained method for model selection - add the pretrained method for model selection

View File

@ -138,7 +138,7 @@ class DeepSAD(BaseDeepAD):
z_ = torch.cat(z_) z_ = torch.cat(z_)
c = torch.mean(z_, dim=0) c = torch.mean(z_, dim=0)
# if c i s too close to zero, set to +- eps # if c is too close to zero, set to +- eps
# a zero unit can be trivially matched with zero weights # a zero unit can be trivially matched with zero weights
c[(abs(c) < eps) & (c < 0)] = -eps c[(abs(c) < eps) & (c < 0)] = -eps
c[(abs(c) < eps) & (c > 0)] = eps c[(abs(c) < eps) & (c > 0)] = eps

View File

@ -21,4 +21,4 @@
# 'X.Y.dev0' is the canonical version of 'X.Y.dev' # 'X.Y.dev0' is the canonical version of 'X.Y.dev'
# #
__version__ = '0.1.1' __version__ = '0.2.0'