s

algorithms/algorithms.py

@@ -6,7 +6,7 @@
  codats_classifier, AdvSKM_Disc
 from models.loss import MMD_loss, CORAL, ConditionalEntropyLoss, VAT, LMMD_loss, HoMM_loss
 from models.augmentations import jitter, scaling, permutation
-
+from torch.optim import SGD
 
 def get_algorithm_class(algorithm_name):
  """Return the algorithm class with the given name."""
@@ -298,7 +298,7 @@ def update(self, src_x, src_y, trg_x, step, epoch, len_train_source, len_train_t
  'open_src_neg_loss': open_loss_neg.item(),
  'open_trg_loss': ent_open.item()
  }
- 
+
 class AdaMatch(Algorithm):
  """
  AdaMatch https://arxiv.org/abs/2106.04732

main_uni.py

@@ -12,9 +12,9 @@
 
 # ======== Experiments Name ================
 parser.add_argument('--save_dir', default='experiments_logs', type=str, help='Directory containing all experiments')
-parser.add_argument('--experiment_description', default='WISDM-TFAC-uni', type=str, help='Name of your experiment (EEG, HAR, HHAR_SA, ')
+parser.add_argument('--experiment_description', default='WISDM-RAINCOAT-uni', type=str, help='Name of your experiment (EEG, HAR, HHAR_SA, ')
 # ========= Select the DA methods ============
-parser.add_argument('--da_method', default='TFAC', type=str, help='DANCE, TFAC')
+parser.add_argument('--da_method', default='RAINCOAT', type=str, help='DANCE, TFAC')
 # ========= Select the DATASET ==============
 parser.add_argument('--data_path', default=r'./data', type=str, help='Path containing dataset')
 parser.add_argument('--dataset', default='WISDM', type=str, help='Dataset of choice: (WISDM - EEG - HAR - HHAR_SA, Boiler)')

models/sinkhorn_vis.py

@@ -1,9 +1,5 @@
 # -*- coding: utf-8 -*-
-"""
-Created on Fri Nov 18 00:46:57 2022
 
-@author: Huan
-"""
 
 import matplotlib.pyplot as plt
 import seaborn as sns