comparison phase2/treeLinkFeatures.m @ 16:4a8ec6c461a0

adding depth checking for recursive tree - feature search - though has a bug where it sometimes get stuck in an infinite loop and I dont know why

-function [linkList, featureList]= treeLinkFeatures(data)
+function [linkList, featureList]= treeLinkFeatures(data, depthThresh)
 %% linkList = treeLinkFeatures(data)
 % given a dataset, a hierarchical cluster of the data is produced, and then
 % the data is traversed, such that, for each split in the data, a set of
 % features are produced, which are the ranked features that can be used to
 % separate the given dataset at that point.
 
 
+
+if(nargin < 2)
+    depthThresh = 9999999;
+end
 linkList = aglomCluster(data);
+linkList = depthCheck(linkList);
 listSize = size(data,1);
 
 % linkList(:,4) = 0;
 featureList = cell(listSize-1,1);
 currentRow = [2*listSize-1];
 
 %%
 while (~isempty(currentRow))
-    if(currentRow(1) > listSize) 
+    if(currentRow(1) > listSize)
         row = currentRow(1) - listSize
-        classList = traceLinkageToBinary(linkList, row);
-        featureList{row} = rfFeatureSelection(data(classList>0,:), classList(classList>0));
-        currentRow = [currentRow(2:end); linkList(row,1); linkList(row,2)];
+        if(linkList(row,4) < depthThresh)
+            classList = traceLinkageToBinary(linkList, row);
+            featureList{row} = rfFeatureSelection(data(classList>0,:), classList(classList>0));
+            currentRow = [currentRow(2:end); linkList(row,1); linkList(row,2)];
+        end
     else
         currentRow = currentRow(2:end);
     end
 end