I’m trying to do transfer learning using ‘Readnet’ on github.
I successfully trained the model by changing,
- Replacing the original train.csv with my own train.csv, in def get_dls(bs):
- Downloading and specifying the glove embeddings (glove.6B.100d.txt), in class GloveEmbedding(nn.Module): and class GloveTokenizer:
- Changing dimensions from d_model=200 to d_model=100, in def get_model():
- Specifying batch size, in learn = Learner(dls=get_dls(32), model=get_model(), loss_func=MSELossFlat())
Please teach me how to get_preds for my texts in the test.csv.
test_df = pd.read_csv('test.csv') learn.dls.test_dl(test_df, 32) learn.get_preds()
just game me (None, None)