facebookresearch/StarSpace

how to structure the labels for training

aartivnkt opened this issue · 0 comments

Hi, I have a very basic question. I'm not sure how to structure the input for training starspace. I have a dataset test.txt with multiple rows like this, is this the correct format?

a<tab>b<tab><c><tab>__label__d<tab>__label__e<tab>__label__f<tab>__label__g<tab>__label__h<tab>__label__i

and for training I'm running

./starspace train -trainFile test.txt -model tagspace -label '__label__'

When I evaluate the model using /.query_predict I'm seeing __label__ in the output. Not sure if I have this right, I appreciate any help!