Line #2.    table = TensorTable.from_df(df.iloc[:100])

can you please add a new cell and add these two lines ( we need my PR to be merged first for the top_k part works ):

Let's now add a top_k parameter to model so that we can return the top k item ids with the highest scores when we serve the model on Triton Inference Server.

model.top_k = 28

model.eval()

Reply via ReviewNB

Line #2. table = TensorTable.from_df(df.iloc[:100])
can you please add a new cell and add these two lines ( we need my PR to be merged first for the top_k part works ):

Let's now add a top_k parameter to model so that we can return the top k item ids with the highest scores when we serve the model on Triton Inference Server.

model.top_k = 28

model.eval()

Reply via ReviewNB

you can set top_k 10 or something else, if you want. I set it to 28 just as an example.