Skip to content

Commit

Permalink
added prediction with classification
Browse files Browse the repository at this point in the history
  • Loading branch information
Xeadriel committed Jun 13, 2024
1 parent c2d535b commit 249a144
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion scripts/fineTune.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -159,7 +159,11 @@ def fineTune(args):

loss = Loss()

logits = predictSentencePairs(model, iterationPairs)
logits = 0
if args.thresholdPrediction:
logits = predictSentencePairsWithDevThresholds(model, iterationPairs, evalPairs, evalLabels)
else:
logits = predictSentencePairs(model, iterationPairs)

lossSim = F.binary_cross_entropy_with_logits(
logits,
Expand Down

0 comments on commit 249a144

Please sign in to comment.