Skip to content

Commit 827438b

Browse files
committed
[SYSTEMDS-3696] Fix incSliceLine flag for pruning strategies
Recent experiments revealed that even with disabled pruning strategies incSliceLine was still faster than sliceLine on some datasets because the reevaluated top-K set was passed to the current top-K set from the beginning and thus used for additional score pruning. We now prevent this if score pruning is disabled.
1 parent 2c7fb4b commit 827438b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

scripts/builtin/incSliceLine.dml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -180,7 +180,7 @@ m_incSliceLine = function(
180180
prevTK2 = matrix(0, 0, ncol(X2));
181181
prevTKC2 = matrix(0, 0, 4);
182182
minsc = -Inf
183-
if( nrow(prevTK) > 0 ) {
183+
if( nrow(prevTK) > 0 & enableIncScorePruning ) {
184184
prevTK2 = oneHotEncodeUsingOffsets(prevTK, foffb, foffe);
185185
[minsc, prevTKC2] = computeLowestPrevTK(prevTK2, X2, totalE, eAvg, alpha)
186186
}

0 commit comments

Comments
 (0)