Adding EnableClassComparison()

joaopauloschuler · Nov 28, 2019 · 1ef4b10 · 1ef4b10
1 parent 9d404f7
commit 1ef4b10
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/neural/neuralfit.pas b/neural/neuralfit.pas
@@ -147,6 +147,7 @@   TNeuralDataLoadingFit = class(TNeuralFitBase)
         Item: TMultiThreadProcItem);
       procedure EnableMonopolarHitComparison();
       procedure EnableBipolarHitComparison();
+      procedure EnableClassComparison();
 
       property DataAugmentationFn: TNNetDataAugmentationFn read FDataAugmentationFn write FDataAugmentationFn;
       property InferHitFn: TNNetInferHitFn read FInferHitFn write FInferHitFn;
@@ -207,6 +208,7 @@   TNeuralImageFit = class(TNeuralFitBase)
 
   function MonopolarCompare(A, B: TNNetVolume): boolean;
   function BipolarCompare(A, B: TNNetVolume): boolean;
+  function ClassCompare(A, B: TNNetVolume): boolean;
 
 implementation
 uses
@@ -250,6 +252,10 @@ function BipolarCompare(A, B: TNNetVolume): boolean;
   end;
 end;
 
+function ClassCompare(A, B: TNNetVolume): boolean;
+begin
+  Result := (A.GetClass() = B.GetClass());
+end;
 { TNeuralDataLoadingFit }
 
 constructor TNeuralDataLoadingFit.Create();
@@ -884,6 +890,11 @@ procedure TNeuralDataLoadingFit.EnableBipolarHitComparison();
   FInferHitFn := @BipolarCompare;
 end;
 
+procedure TNeuralDataLoadingFit.EnableClassComparison();
+begin
+  FInferHitFn := @ClassCompare;
+end;
+
 { TNeuralFitBase }
 
 constructor TNeuralFitBase.Create();