update the regulation term

DravenALG · Sep 4, 2023 · dc428b7 · dc428b7
1 parent 51dcb4c
commit dc428b7
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/MAIN.m b/MAIN.m
@@ -2,7 +2,7 @@
 close all; clear; clc; warning off;
 
 %% globel settings
-train_param.current_bits = 16;
+train_param.current_bits = 128;
 train_param.max_iter=5;
 
 %% load dataset

diff --git a/train_OASIS.m b/train_OASIS.m
@@ -39,7 +39,7 @@
 
 
  % update U
- U_new = (GTrain_new*V_new'+HH{1,2})/(V_new*V_new'+HH{1,3}+thet);
+ U_new = (GTrain_new*V_new'+HH{1,2})/(V_new*V_new'+HH{1,3}+thet*eye(nbits));
 
  % update W
  WW{1,1}=(B_new*Xm1'+HH{1,5})/(Xm1*Xm1'+HH{1,6}+delta*eye(param.image_feature_size));

diff --git a/train_OASIS0.m b/train_OASIS0.m
@@ -39,7 +39,7 @@
 
 
  % update U
- U_new = (GTrain_new*V_new')/(V_new*V_new'+thet);
+ U_new = (GTrain_new*V_new')/(V_new*V_new'+thet*eye(nbits));
 
  % update W
  WW{1,1}=B_new*Xm1'/(Xm1*Xm1'+delta*eye(param.image_feature_size));