From 6b3a41448c9c89b643b5990bdbf93c84c4ec249f Mon Sep 17 00:00:00 2001 From: Jan Kowalczyk Date: Wed, 2 Jul 2025 16:58:34 +0200 Subject: [PATCH] also save linear ocsvm model --- Deep-SAD-PyTorch/src/baselines/ocsvm.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/Deep-SAD-PyTorch/src/baselines/ocsvm.py b/Deep-SAD-PyTorch/src/baselines/ocsvm.py index fe6ca55..5b2f98b 100644 --- a/Deep-SAD-PyTorch/src/baselines/ocsvm.py +++ b/Deep-SAD-PyTorch/src/baselines/ocsvm.py @@ -195,9 +195,7 @@ class OCSVM(object): # If hybrid, also train a model with linear kernel if self.hybrid: - self.linear_model = OneClassSVM( - kernel="linear", nu=self.nu - ) + self.linear_model = OneClassSVM(kernel="linear", nu=self.nu) start_time = time.time() self.linear_model.fit(X) train_time = time.time() - start_time @@ -480,12 +478,14 @@ class OCSVM(object): def save_model(self, export_path): """Save OC-SVM model to export_path.""" with open(export_path, "wb") as fp: - pickle.dump(self.model, fp) + pickle.dump({"model": self.model, "linear_model": self.linear_model}, fp) def load_model(self, import_path): """Load OC-SVM model from import_path.""" with open(import_path, "rb") as fp: - self.model = pickle.load(fp) + models = pickle.load(fp) + self.model = models["model"] + self.linear_model = models["linear_model"] def save_results(self, export_pkl): with open(export_pkl, "wb") as fp: