Skip to content

Commit

Permalink
Update
Browse files Browse the repository at this point in the history
  • Loading branch information
olegranmo committed Aug 17, 2024
1 parent cf1956d commit 5319b4e
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 17 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -115,7 +115,7 @@ def main(args):
X_test[i,0,position_5,0] = 0
X_test[i,0,position_5,1] = 0

tm = TMClassifier(args.number_of_clauses, args.T, args.s, number_of_state_bits_ta=10, patch_dim=(1, 1), weighted_clauses=True, platform=args.platform, boost_true_positive_feedback=True, spatio_temporal=True, incremental=False, max_included_literals=args.max_included_literals)
tm = TMClassifier(args.number_of_clauses, args.T, args.s, number_of_state_bits_ta=10, patch_dim=(1, 1), weighted_clauses=False, platform=args.platform, boost_true_positive_feedback=True, spatio_temporal=True, incremental=False, max_included_literals=args.max_included_literals)

for i in range(args.epochs):
tm.fit(X_train, Y_train)
Expand Down
16 changes: 0 additions & 16 deletions tmu/lib/src/ClauseBank.c
Original file line number Diff line number Diff line change
Expand Up @@ -1538,14 +1538,6 @@ void cb_calculate_spatio_temporal_features(
chunk_nr = (number_of_clauses*4*d + j + number_of_literals/2) / 32;
chunk_pos = (number_of_clauses*4*d + j + number_of_literals/2) % 32;
Xi[(patch-1)*number_of_ta_chunks + chunk_nr] &= ~(1U << chunk_pos);
} else {
chunk_nr = (number_of_clauses*4*d + j) / 32;
chunk_pos = (number_of_clauses*4*d + j) % 32;
Xi[(patch-1)*number_of_ta_chunks + chunk_nr] &= ~(1U << chunk_pos);

chunk_nr = (number_of_clauses*4*d + j + number_of_literals/2) / 32;
chunk_pos = (number_of_clauses*4*d + j + number_of_literals/2) % 32;
Xi[(patch-1)*number_of_ta_chunks + chunk_nr] |= (1U << chunk_pos);
}
}

Expand All @@ -1559,14 +1551,6 @@ void cb_calculate_spatio_temporal_features(
chunk_nr = (number_of_clauses*4*d + j + number_of_clauses + number_of_literals/2) / 32;
chunk_pos = (number_of_clauses*4*d + j + number_of_clauses + number_of_literals/2) % 32;
Xi[(patch+1)*number_of_ta_chunks + chunk_nr] &= ~(1U << chunk_pos);
} else {
chunk_nr = (number_of_clauses*4*d + j + number_of_clauses) / 32;
chunk_pos = (number_of_clauses*4*d + j + number_of_clauses) % 32;
Xi[(patch+1)*number_of_ta_chunks + chunk_nr] &= ~(1U << chunk_pos);

chunk_nr = (number_of_clauses*4*d + j + number_of_clauses + number_of_literals/2) / 32;
chunk_pos = (number_of_clauses*4*d + j + number_of_clauses + number_of_literals/2) % 32;
Xi[(patch+1)*number_of_ta_chunks + chunk_nr] |= (1U << chunk_pos);
}
}

Expand Down

0 comments on commit 5319b4e

Please sign in to comment.