forked from JointCloud/JCC-DeepOD
commit to merge
This commit is contained in:
parent
1af6bcc48d
commit
efc80c1e97
1
TODO.md
1
TODO.md
|
@ -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
|
||||||
|
|
|
@ -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
|
||||||
|
|
|
@ -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'
|
||||||
|
|
Loading…
Reference in New Issue