Matlab:将用户 - 项目 - 评级CSV加载到评级矩阵中

时间:2017-07-31 22:44:19

标签: matlab csv matrix recommendation-engine

我对用户评分项目有以下CSV格式:

A1YS,8F20,3.0
A3TS,8320,2.0
A3BU,1905,5.0
A3BU,3574,4.0
A14X,185A,1.0

列是UserID,ItemID,Rating

我想将其加载到带有用户行的Matlab矩阵中,项目列和单元格值将包含评级(未知等级等于零),例如:

      8F20,  1905,  3574,  185A
A1YS    3 ,    0 ,    0 ,    0 
A3TS    2 ,    0 ,    0 ,    0 
A3BU    0 ,    5 ,    4 ,    0 
A14X    0 ,    0 ,    0 ,    1 

另一件事,实际上矩阵可以形成为:

3 ,    0 ,    0 ,    0 
2 ,    0 ,    0 ,    0 
0 ,    5 ,    4 ,    0 
0 ,    0 ,    0 ,    1 

我对Matlab很新,并尝试了一些变体:

https://stackoverflow.com/a/13775907/1726419https://stackoverflow.com/a/19613301/1726419

没有大的成功 - 我会非常感谢任何帮助。

修改: 到目前为止我得到的是:

fid = fopen('ratings_sample.csv');
out = textscan(fid,'%s%s%d%d','delimiter',',');
fclose(fid);

c1 = out{1};
c2 = out{2};
c3 = out{3};

我的问题是我需要重复删除c1和& c2并正确填写矩阵的内部单元格。另外,我不知道这是否是加载它的正确方法。

1 个答案:

答案 0 :(得分:2)

如果UserID和ItemID是唯一的,您可以使用crosstab

UserID = categorical(c1);
ItemID = categorical(c2);
Rating = crosstab(UserID,ItemID);
Rating(Rating==1) = c3;

并获得:

Rating =
     3     0     0     0     0
     0     0     0     0     1
     0     2     5     0     0
     0     0     0     4     0

如果要在表格中对其进行整理,则需要先将项目ID转换为有效的变量名称(以字母开头):

Items = cellfun(@(s) ['Item_' s],c2,'un',0);

然后您可以使用table来保存所有数据:

Tbl = array2table(Rating,...
                  'RowNames',unique(c1,'stable'),...
                  'VariableNames',unique(Items,'stable'))

结果:

Tbl =
  4×5 table
            Item_8F20    Item_8320    Item_1905    Item_3574    Item_185A
            _________    _________    _________    _________    _________
    A1YS    3            0            0            0            0        
    A3TS    0            0            0            0            1        
    A3BU    0            2            5            0            0        
    A14X    0            0            0            4            0