cmfrec 0.1.6 → 0.1.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 34e8dc08914cbc418470cd7eb3adf3d33013b786319f4510212c80bf3629f3ca
-  data.tar.gz: c1b91a1f77b4b51a5ca4491376f8a02230ea54873f8c1b2b06f4761d6ddd0686
+  metadata.gz: 8d16ab98cb7de22042eaf353a9d41d0b7a4214a631a373c553f73825418c026a
+  data.tar.gz: 9ab678a9d389b835b4dfd91d14c372d5acfef950bf068ac46d2d879af04f0fcc
 SHA512:
-  metadata.gz: a3c57734379199196a4e3f51d9ec02b19ef1abac13d57a10ca3c20e9b76c9ee5db4b17d790330d41a9576c2ba28a9eeccafeb5760b54cfdf80a7431368895068
-  data.tar.gz: 5a24a77a6665854abb38916a22e8141a6cae637a51f98e3df3762566f2e73cb60b9bd9a25303df0411ea53dec3211bf7534711dc55c510311620341cbe4e4ac3
+  metadata.gz: 175d3c91056d2e8734af6961471c98be76e4d5f6b85faaecdfd3b39a220efafa70150983e9d74efb1a1211a29e6c867d6b1d7f482cc34c55500268d29b40158c
+  data.tar.gz: faaed621391ccc7d94f2e6309481b24f7db62fbe86eb6c1bbb35445dde0cabf32c41791af4ebcd25e530bdc1f6319f1b4477b6c4d6ca1bf77353d3a0c4ae8d5c

data/CHANGELOG.md

@@ -1,3 +1,9 @@
+## 0.1.7 (2022-03-22)
+
+- Improved ARM detection
+- Fixed error with `load_movielens`
+- Fixed duplicates in `item_info` with `load_movielens`
+
 ## 0.1.6 (2021-08-12)
 
 - Added `user_ids` and `item_ids` methods

data/README.md

@@ -1,4 +1,4 @@
-# cmfrec
+# cmfrec Ruby
 
 :fire: Recommendations for Ruby, powered by [cmfrec](https://github.com/david-cortes/cmfrec)
 
@@ -6,7 +6,7 @@
 - Works with explicit and implicit feedback
 - Uses high-performance matrix factorization
 
-[![Build Status](https://github.com/ankane/cmfrec/workflows/build/badge.svg?branch=master)](https://github.com/ankane/cmfrec/actions)
+[![Build Status](https://github.com/ankane/cmfrec-ruby/workflows/build/badge.svg?branch=master)](https://github.com/ankane/cmfrec-ruby/actions)
 
 ## Installation
 
@@ -58,8 +58,8 @@ Get recommendations for a new user
 
 ```ruby
 recommender.new_user_recs([
-  {item_id: 1, value: 5},
-  {item_id: 2, value: 3}
+  {item_id: 1, rating: 5},
+  {item_id: 2, rating: 3}
 ])
 ```
 
@@ -150,11 +150,7 @@ recommender.predict(ratings.last(20000))
 [Ahoy](https://github.com/ankane/ahoy) is a great source for implicit feedback
 
 ```ruby
-views = Ahoy::Event.
-  where(name: "Viewed post").
-  group(:user_id).
-  group("properties->>'post_id'"). # postgres syntax
-  count
+views = Ahoy::Event.where(name: "Viewed post").group(:user_id).group_prop(:post_id).count
 
 data =
   views.map do |(user_id, post_id), count|
@@ -230,6 +226,8 @@ bin = File.binread("recommender.bin")
 recommender = Marshal.load(bin)
 ```
 
+Alternatively, you can store only the factors and use a library like [Neighbor](https://github.com/ankane/neighbor). See the [examples](https://github.com/ankane/neighbor/tree/master/examples) for Disco, which has a similar API. For explicit feedback, you should [disable the bias](#explicit-feedback) with this approach.
+
 ## Reference
 
 Get ids
@@ -269,22 +267,22 @@ Cmfrec.ffi_lib = "path/to/cmfrec.dll"
 
 ## History
 
-View the [changelog](https://github.com/ankane/cmfrec/blob/master/CHANGELOG.md)
+View the [changelog](https://github.com/ankane/cmfrec-ruby/blob/master/CHANGELOG.md)
 
 ## Contributing
 
 Everyone is encouraged to help improve this project. Here are a few ways you can help:
 
-- [Report bugs](https://github.com/ankane/cmfrec/issues)
-- Fix bugs and [submit pull requests](https://github.com/ankane/cmfrec/pulls)
+- [Report bugs](https://github.com/ankane/cmfrec-ruby/issues)
+- Fix bugs and [submit pull requests](https://github.com/ankane/cmfrec-ruby/pulls)
 - Write, clarify, or fix documentation
 - Suggest or add new features
 
 To get started with development:
 
 ```sh
-git clone https://github.com/ankane/cmfrec.git
-cd cmfrec
+git clone https://github.com/ankane/cmfrec-ruby.git
+cd cmfrec-ruby
 bundle install
 bundle exec rake vendor:all
 bundle exec rake test

data/lib/cmfrec/data.rb

@@ -3,11 +3,11 @@ module Cmfrec
     def load_movielens
       require "csv"
 
-      data_path = download_file("ml-100k/u.data", "http://files.grouplens.org/datasets/movielens/ml-100k/u.data",
+      data_path = download_file("ml-100k/u.data", "https://files.grouplens.org/datasets/movielens/ml-100k/u.data",
         file_hash: "06416e597f82b7342361e41163890c81036900f418ad91315590814211dca490")
-      user_path = download_file("ml-100k/u.user", "http://files.grouplens.org/datasets/movielens/ml-100k/u.user",
+      user_path = download_file("ml-100k/u.user", "https://files.grouplens.org/datasets/movielens/ml-100k/u.user",
         file_hash: "f120e114da2e8cf314fd28f99417c94ae9ddf1cb6db8ce0e4b5995d40e90e62c")
-      item_path = download_file("ml-100k/u.item", "http://files.grouplens.org/datasets/movielens/ml-100k/u.item",
+      item_path = download_file("ml-100k/u.item", "https://files.grouplens.org/datasets/movielens/ml-100k/u.item",
         file_hash: "553841ebc7de3a0fd0d6b62a204ea30c1e651aacfb2814c7a6584ac52f2c5701")
 
       # convert u.item to utf-8
@@ -24,8 +24,13 @@ module Cmfrec
 
       item_info = []
       movies = {}
+      movie_names = {}
       genres = %w(unknown action adventure animation childrens comedy crime documentary drama fantasy filmnoir horror musical mystery romance scifi thriller war western)
       CSV.parse(movies_str, col_sep: "|", converters: [:numeric]) do |row|
+        # filter duplicates
+        next if movie_names[row[1]]
+        movie_names[row[1]] = true
+
         movies[row[0]] = row[1]
         item = {item_id: row[1], year: row[2] ? Date.parse(row[2]).year : 1970}
         genres.each_with_index do |genre, i|
@@ -49,7 +54,10 @@ module Cmfrec
     private
 
     def download_file(fname, origin, file_hash:)
+      require "digest"
       require "fileutils"
+      require "net/http"
+      require "tmpdir"
 
       # TODO handle this better
       raise "No HOME" unless ENV["HOME"]
@@ -58,10 +66,6 @@ module Cmfrec
 
       return dest if File.exist?(dest)
 
-      require "digest"
-      require "net/http"
-      require "tmpdir"
-
       temp_path = "#{Dir.tmpdir}/cmfrec-#{Time.now.to_f}" # TODO better name
 
       digest = Digest::SHA2.new

data/lib/cmfrec/version.rb

@@ -1,3 +1,3 @@
 module Cmfrec
-  VERSION = "0.1.6"
+  VERSION = "0.1.7"
 end

data/lib/cmfrec.rb

@@ -19,7 +19,7 @@ module Cmfrec
     if Gem.win_platform?
       "cmfrec.dll"
     elsif RbConfig::CONFIG["host_os"] =~ /darwin/i
-      if RbConfig::CONFIG["host_cpu"] =~ /arm/i
+      if RbConfig::CONFIG["host_cpu"] =~ /arm|aarch64/i
         "libcmfrec.arm64.dylib"
       else
         "libcmfrec.dylib"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: cmfrec
 version: !ruby/object:Gem::Version
-  version: 0.1.6
+  version: 0.1.7
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-08-12 00:00:00.000000000 Z
+date: 2022-03-22 00:00:00.000000000 Z
 dependencies: []
 description:
 email: andrew@ankane.org
@@ -28,7 +28,7 @@ files:
 - vendor/libcmfrec.arm64.dylib
 - vendor/libcmfrec.dylib
 - vendor/libcmfrec.so
-homepage: https://github.com/ankane/cmfrec
+homepage: https://github.com/ankane/cmfrec-ruby
 licenses:
 - MIT
 metadata: {}
@@ -47,7 +47,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.2.22
+rubygems_version: 3.3.7
 signing_key:
 specification_version: 4
 summary: Recommendations for Ruby using collective matrix factorization