a/ebooks
1
1
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Take top 5 instead of top 3

Browse source
This commit is contained in:
Anthony Wang 2022-02-21 16:39:58 -06:00
parent f2d33b51b1
commit b240317d76
Signed by: a
GPG key ID: BC96B00AEC5F2D76
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
predict.py
View file

@ -2,7 +2,7 @@ import numpy as np
import torch
def predict(device, dataset, model, text, next_words=100, top_k=3):
def predict(device, dataset, model, text, next_words=100, top_k=5):
model.eval()
words = text.split()