GenerTeam commited on
Commit
fabe9ba
·
verified ·
1 Parent(s): 9a20dad

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -161,11 +161,11 @@ bp_probs_list, actual_probs_list = model.score_sequence(sequences)
161
 
162
  # Compute metrics for each sequence
163
  for i, (seq, actual_probs) in enumerate(zip(sequences, actual_probs_list)):
164
- log_likelihood = actual_probs.log().sum().item() # Total log-likelihood
165
  perplexity = torch.exp(-actual_probs.log().mean()).item() # Perplexity
166
 
167
  print(f"Sequence {i+1} (length {len(seq)}):")
168
- print(f" Log-likelihood: {log_likelihood:.2f}")
169
  print(f" Perplexity: {perplexity:.4f}")
170
  print(f" Mean probability: {actual_probs.mean().item():.4f}")
171
  ```
 
161
 
162
  # Compute metrics for each sequence
163
  for i, (seq, actual_probs) in enumerate(zip(sequences, actual_probs_list)):
164
+ log_likelihood = actual_probs.log().mean().item() # Total log-likelihood
165
  perplexity = torch.exp(-actual_probs.log().mean()).item() # Perplexity
166
 
167
  print(f"Sequence {i+1} (length {len(seq)}):")
168
+ print(f" Mean log-likelihood: {log_likelihood:.2f}")
169
  print(f" Perplexity: {perplexity:.4f}")
170
  print(f" Mean probability: {actual_probs.mean().item():.4f}")
171
  ```