fix: Take into account only non empty cell

RomualdRousseau · Nov 13, 2024 · 5df17c7 · 5df17c7
1 parent aad0b13
commit 5df17c7
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/archery/src/main/java/com/github/romualdrousseau/archery/header/DataTableHeader.java b/archery/src/main/java/com/github/romualdrousseau/archery/header/DataTableHeader.java
@@ -83,17 +83,25 @@ private List<String> sampleEntities() {
         final var N = Math.min(this.getTable().getNumberOfRows(), Settings.DEFAULT_SAMPLE_COUNT);
         final var entityVector = Tensor
                 .zeros(this.getTable().getSheet().getDocument().getModel().getEntityList().size());
+        float n = 0.0f;
         for (int i = 0; i < N; i++) {
             final BaseRow row = this.getTable().getRowAt(i);
             if (row == null) {
                 continue;
             }
+
             final BaseCell cell = row.getCellAt(this.getColumnIndex());
-            if (cell.hasValue() && cell.getSymbol().equals("e")) {
+            if (!cell.hasValue()) {
+                continue;
+            }
+
+            if (cell.getSymbol().equals("e")) {
                 entityVector.iadd(cell.getEntityVector());
             }
+
+            n += Settings.DEFAULT_ENTITY_PROBABILITY;
         }
-        entityVector.if_lt_then(N * Settings.DEFAULT_ENTITY_PROBABILITY, 0.0f, 1.0f);
+        entityVector.if_lt_then(n, 0.0f, 1.0f);
         final var entityList = this.getTable().getSheet().getDocument().getModel().getEntityList();
         return IntStream.range(0, entityVector.size).boxed()
                 .filter(i -> entityVector.data[i] == 1)