|
Algorithm 1 FW-RFE |
Input: The set of output features of the pooling layer and sample labels,
and the set of convolutional layer feature weights,
Output: Feature Sorted Set R;
-
1:
Initial feature set S
and accuracy set ;
-
2:
whiledo
-
3:
Calculate the ranking criterion score:;
-
4:
Find the feature with the smallest ranking score:
-
5:
Extract features , updata S;
-
6:
Take S as the input of the fully connected layer, use sigmoid regression function to achieve binary classification:
-
7:
Update accuracy set:
-
8:
end while
-
9:
When the accuracy rate in the accuracy set V is the highest, the feature set in S is selected as the output result
|