Building datasets

Hello !

Thank you for this open source package, it help a lot and your work is amazing.

I just a have a silly question about dataset construction. I followed the example for my data: user (160.000 x 300) and item (4000 x 4).

dataset = Dataset()
dataset.fit(users=(x['id_user'] for x in user),
            items=(x['id_item'] for x in item),
            user_features=((x['id_user'], [[x[col] for col in list_columns_user]]) for x in user),
            item_features=((x['id_item'], [[x[col] for col in list_columns_item]]) for x in item))

But when I try dataset.user_features_shape() I get (160000, 160000). shouldn’t I rather have this (160000, 300) ?

Indeed, we can read in the documentation :

Returns ------- (num user ids, num user features): tuple of ints

and my num user features is 300. So there is an error in what I did?

Sorry for the stupid question!

[maciejkula]

You need to pass an iterable of tuples of (id, [list of features for that id]) into build_features. It looks like at the moment you’re passing the same features for every user?

[maciejkula]

(Well, you should get 160000 x 1600300 or something like that. Are your feature names the same as some of your user ids?)