fix gpr training scaling for CovYI

aerialhedgehog · Jun 4, 2015 · ac2a705 · ac2a705
1 parent de0e258
commit ac2a705
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/trunk/VyPy/regression/gpr/scaling/Linear.py b/trunk/VyPy/regression/gpr/scaling/Linear.py
@@ -36,6 +36,7 @@ def calc_scaling(self,Train):
         X_scaling  = LinearFunction( Xref[None,0,:] , Xref[None,1,:] )
         Y_scaling  = LinearFunction( Yref[0], Yref[1] )
         DY_scaling = LinearFunction( DYref[None,0,:]  )
+        CovYI_scaling  = LinearFunction( Yref[0] )
 
         # set scaling data keys
         self['XB']     = XB_scaling
@@ -45,7 +46,7 @@ def calc_scaling(self,Train):
         self['XI']     = X_scaling
         self['YI']     = Y_scaling
         self['DYI']    = DY_scaling
-        self['CovYI']  = Y_scaling
+        self['CovYI']  = CovYI_scaling
         self['CovDYI'] = DY_scaling
 
         return