cumf/cumf_sgd

CUDA Matrix Factorization Library with Stochastic Gradient Descent (SGD)

cumf_sgd's Issues

setting parameters

Hello
I want to run cumfsgd on Yahoo! Music dataset.
How should I set the parameters to get the result?

license

Can you please add a license to your repository ?

Thanks

The netflix dataset is not available.

Are there any alternative links?

How to extract X and Theta

How and where do I extract X and Theta after convergence? Are these being written to a file on the host device?

Write conflicts in the update procedure.

Hi, I think there is a bug in the 89-99 lines of cumf_sgd/singleGPU/sgd_k128_kernel_hogwild_warp32.h:

//update
//only works for k=blockDim.x=128
p[base_p + lane_id +  0] = __float2half(tmp_p1 + tmp_lrate*(ruv*tmp_q1 - lambda_p*tmp_p1));
q[base_q + lane_id +  0] = __float2half(tmp_q1 + tmp_lrate*(ruv*tmp_p1 - lambda_q*tmp_q1));

p[base_p + lane_id + 32] = __float2half(tmp_p2 + tmp_lrate*(ruv*tmp_q2 - lambda_p*tmp_p2));
q[base_q + lane_id + 32] = __float2half(tmp_q2 + tmp_lrate*(ruv*tmp_p2 - lambda_q*tmp_q2));

p[base_p + lane_id + 64] = __float2half(tmp_p3 + tmp_lrate*(ruv*tmp_q3 - lambda_p*tmp_p3));
q[base_q + lane_id + 64] = __float2half(tmp_q3 + tmp_lrate*(ruv*tmp_p3 - lambda_q*tmp_q3));

p[base_p + lane_id + 96] = __float2half(tmp_p4 + tmp_lrate*(ruv*tmp_q4 - lambda_p*tmp_p4));
q[base_q + lane_id + 96] = __float2half(tmp_q4 + tmp_lrate*(ruv*tmp_p4 - lambda_q*tmp_q4));

Multiple workers(blocks) or warps may update the same base_p or base_q so that there will be write conflicts. This procedure should use atomic operations.

netflix_mm/netflix_mme file

Hi,

I'm doing some unrelated research, resurrecting a project from a couple years ago. It relies on the netflix_mm and netflix_mme files, which seem to have completely disappeared from the Internet. Your README does mention these files as being in your google drive link, but I can't seem to find them (you have netflix_train and netflix_test, but these appear to be in a different format).

Do you, by any chance, have the netflix_mm and netflix_mme files, and would it be at all possible for you to (temporarily) upload them somewhere where I could download them? It would be much appreciated.

Thanks in advance!