cofi_cost 0.0.3 → 0.0.4
Sign up to get free protection for your applications and to get access to all the features.
- data/lib/cofi_cost.rb +10 -7
- metadata +5 -5
data/lib/cofi_cost.rb
CHANGED
|
@@ -6,8 +6,8 @@ include GSL::MultiMin
|
|
|
6
6
|
|
|
7
7
|
class CofiCost
|
|
8
8
|
|
|
9
|
-
attr_accessor :ratings, :num_features, :cost, :lambda, :iterations
|
|
10
|
-
attr_reader :boolean_rated, :num_tracks, :num_users, :
|
|
9
|
+
attr_accessor :ratings, :num_features, :cost, :lambda, :iterations, :features, :theta
|
|
10
|
+
attr_reader :boolean_rated, :num_tracks, :num_users, :ratings_mean, :ratings_norm, :predictions
|
|
11
11
|
|
|
12
12
|
def initialize(ratings, num_features = 2, lambda = 1, iterations = 10, features = nil, theta = nil)
|
|
13
13
|
@ratings = ratings.to_f # make sure it's a float for correct normalization
|
|
@@ -33,7 +33,12 @@ class CofiCost
|
|
|
33
33
|
for i in 0..@num_tracks-1 # sadly, @num_tracks.each_index does not work with NArray yet
|
|
34
34
|
track_rating = @ratings[true,i] # get all user ratings for track i (including unrated)
|
|
35
35
|
boolean_track_rating = boolean_rated[true,i] # get all user ratings that exist for track i
|
|
36
|
-
|
|
36
|
+
track_rating_boolean = track_rating[boolean_track_rating]
|
|
37
|
+
if track_rating_boolean.size == 0
|
|
38
|
+
@ratings_mean[i] = 0
|
|
39
|
+
else
|
|
40
|
+
@ratings_mean[i] = track_rating_boolean.mean
|
|
41
|
+
end
|
|
37
42
|
|
|
38
43
|
track_norm = @ratings_norm[true,i]
|
|
39
44
|
track_norm[boolean_track_rating] = track_rating[boolean_track_rating] - @ratings_mean[i]
|
|
@@ -154,10 +159,8 @@ class NArray
|
|
|
154
159
|
end
|
|
155
160
|
|
|
156
161
|
##ratings = NArray.float(4,5).indgen(0,2)
|
|
157
|
-
#ratings = NArray[[
|
|
158
|
-
#
|
|
159
|
-
#lambda = 1
|
|
160
|
-
#g = CofiCost.new(ratings, num_features, lambda)
|
|
162
|
+
#ratings = NArray[[1.0,4.0],[4.0,0.0],[0.0,0.0]]
|
|
163
|
+
#g = CofiCost.new(ratings, 5, 1, 10, nil, nil)
|
|
161
164
|
#puts g.theta.nil?
|
|
162
165
|
#g.min_cost
|
|
163
166
|
#puts "new theta"
|
metadata
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
--- !ruby/object:Gem::Specification
|
|
2
2
|
name: cofi_cost
|
|
3
3
|
version: !ruby/object:Gem::Version
|
|
4
|
-
version: 0.0.
|
|
4
|
+
version: 0.0.4
|
|
5
5
|
prerelease:
|
|
6
6
|
platform: ruby
|
|
7
7
|
authors:
|
|
@@ -13,7 +13,7 @@ date: 2012-09-02 00:00:00.000000000 Z
|
|
|
13
13
|
dependencies:
|
|
14
14
|
- !ruby/object:Gem::Dependency
|
|
15
15
|
name: gsl
|
|
16
|
-
requirement: &
|
|
16
|
+
requirement: &77565430 !ruby/object:Gem::Requirement
|
|
17
17
|
none: false
|
|
18
18
|
requirements:
|
|
19
19
|
- - ! '>='
|
|
@@ -21,10 +21,10 @@ dependencies:
|
|
|
21
21
|
version: '0'
|
|
22
22
|
type: :runtime
|
|
23
23
|
prerelease: false
|
|
24
|
-
version_requirements: *
|
|
24
|
+
version_requirements: *77565430
|
|
25
25
|
- !ruby/object:Gem::Dependency
|
|
26
26
|
name: narray
|
|
27
|
-
requirement: &
|
|
27
|
+
requirement: &77565210 !ruby/object:Gem::Requirement
|
|
28
28
|
none: false
|
|
29
29
|
requirements:
|
|
30
30
|
- - ! '>='
|
|
@@ -32,7 +32,7 @@ dependencies:
|
|
|
32
32
|
version: '0'
|
|
33
33
|
type: :runtime
|
|
34
34
|
prerelease: false
|
|
35
|
-
version_requirements: *
|
|
35
|
+
version_requirements: *77565210
|
|
36
36
|
description: Playground for collaborative filtering in Ruby using NArray and rb-gsl.
|
|
37
37
|
email: tomwolfe@gmail.com
|
|
38
38
|
executables: []
|