fedex/mt
Public Access
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ocsvm working

Browse Source
This commit is contained in:
Jan Kowalczyk
2025-06-13 10:24:54 +02:00
parent d88719e718
commit 9298dea329
6 changed files with 376 additions and 137 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
Deep-SAD-PyTorch/src/networks/main.py
View File

@@ -9,7 +9,7 @@ from .subter_LeNet_Split import SubTer_LeNet_Split, SubTer_LeNet_Split_Autoencod
from .vae import VariationalAutoencoder
def build_network(net_name, ae_net=None, rep_dim=1024):
def build_network(net_name, rep_dim, ae_net=None):
"""Builds the neural network."""
implemented_networks = (
@@ -129,7 +129,7 @@ def build_network(net_name, ae_net=None, rep_dim=1024):
return net
def build_autoencoder(net_name):
def build_autoencoder(net_name, rep_dim):
"""Builds the corresponding autoencoder network."""
implemented_networks = (
@@ -158,7 +158,7 @@ def build_autoencoder(net_name):
ae_net = MNIST_LeNet_Autoencoder()
if net_name == "subter_LeNet":
ae_net = SubTer_LeNet_Autoencoder()
ae_net = SubTer_LeNet_Autoencoder(rep_dim=rep_dim)
if net_name == "subter_LeNet_Split":
ae_net = SubTer_LeNet_Split_Autoencoder()