Bug fix using raw input data in the model weights

peterson-tim-j · Feb 13, 2024 · 71ab7b5 · 71ab7b5
1 parent ad9937d
commit 71ab7b5
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/algorithms/models/TransferNoise/model_TFN.m b/algorithms/models/TransferNoise/model_TFN.m
@@ -1152,6 +1152,16 @@
             forcingData_colnames = forcingData_colnames(filt);
             setForcingData(obj, forcingData_data, forcingData_colnames);
 
+            % Update forcing data column numbers for those components using
+            % the raw inoput data. 
+            filt = find(strcmpi(varargin(:,2),'forcingdata'));
+            for ii=filt'
+               modelComponent = varargin{ii,1};
+               if ischar(varargin{ii,3})
+                  filt_forcingCols = cellfun(@(x,y)(strcmp(x,varargin{ii,3})), forcingData_colnames);
+                  obj.inputData.componentData.(modelComponent).dataColumn  = find(filt_forcingCols);
+               end
+            end
             %-------------
         end