search

tlatkowski / multihead-siamese-nets

Implementation of Siamese Neural Networks built upon multihead attention mechanism for text semantic similarity task.
MIT License
182 stars 43 forks source link

Train a model using a new dataset #24

Open abrahami opened 4 years ago

abrahami commented 4 years ago

hi there,

thanks a lot for the great work here! Is it possible to train a model using a new dataset (not one out of the three suggested in the repo)?

thanks!

power76 commented 4 years ago

Sure, but your dataset must comply with the form of SNLI or the other two.

abrahami commented 4 years ago

Great, can you elaborate a bit more and guide me how to do it? Seems like the input can only get and handle the specific 3 datasets you are pointing to

thanks