Wrong output

[limapedro]

When I try to predict some sentence I get the prediction for each word.
Is this a bug?

[giacbrd]

Hi, can you give me an example of input sentence?

[limapedro]

Here my source:

from shallowlearn.models import GensimFastText

clf = GensimFastText(size=100, min_count=0, loss='hs', iter=1, seed=66)'''

clf.fit([('i', 'am', 'tall'), ('you', 'are', 'fat')], ['yes', 'no'])
print(clf.predict(['tall', 'am', 'i']))

It outputs: ['no', 'no', 'no'] Instead of just 'no'

[giacbrd]

This is because predict takes a list of samples, while you are passing a single sample: ['tall', 'am', 'i']
Try for example clf.predict([('tall', 'am', 'i')])