From 2b939e00794e2e2bdc1566d83cfd7c14038ded56 Mon Sep 17 00:00:00 2001
From: Baptiste Bauvin <baptiste.bauvin@lis-lab.fr>
Date: Wed, 15 Mar 2023 06:50:23 -0400
Subject: [PATCH] SPKM update
---
summit/multiview_platform/monoview_classifiers/spkm.py | 2 +-
summit/multiview_platform/multiview_classifiers/spkm_pw.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/summit/multiview_platform/monoview_classifiers/spkm.py b/summit/multiview_platform/monoview_classifiers/spkm.py
index 3d710131..98b72130 100644
--- a/summit/multiview_platform/monoview_classifiers/spkm.py
+++ b/summit/multiview_platform/monoview_classifiers/spkm.py
@@ -33,7 +33,7 @@ class SPKM(BaseMonoviewClassifier, SPKMl2l1likeSklearn):
self.distribs = [[2], [PolyKernel({"d":3, "r":1}), RBFKernel(0.5)],
CustomUniform(),
["data"], [10],
- [[0,1],], ["l1ball", "proxgd"]
+ [[0,1],[0,1,2]], ["l1ball", "proxgd"]
[random_state],]
self.more_than_two_views = False
self.weird_strings = []
diff --git a/summit/multiview_platform/multiview_classifiers/spkm_pw.py b/summit/multiview_platform/multiview_classifiers/spkm_pw.py
index 1d791e79..b6cdc0a9 100644
--- a/summit/multiview_platform/multiview_classifiers/spkm_pw.py
+++ b/summit/multiview_platform/multiview_classifiers/spkm_pw.py
@@ -35,7 +35,7 @@ class PWSPKM(BaseMultiviewClassifier, pairwiseSPKMl2l1likeSklearn):
[PolyKernel({"d":3, "r":1}), RBFKernel(0.5)],
CustomUniform(),
["data"], [10],
- [[0,1],], ["l1ball", "proxgd"]
+ [[0,1],[0,1,2]], ["l1ball", "proxgd"]
[random_state],]
self.more_than_two_views = False
self.random_state = random_state
--
GitLab