irt_ruby 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 3864e3faec7cbf0a1d96649b1493a843df0fcb029a2106b99e682f1770b9f499
+  data.tar.gz: e6901babf0c08c5de4e8d644ee4c5e8607f1213a965864874455af2e1ea5ead5
+SHA512:
+  metadata.gz: 519eb1b5d1e5c74234482b83d72faacb85445b5259860278512a49f3212ce62840a8db8c801d6873be377b7b6cf808693af4456185bf2a64a679b20b6f21a33b
+  data.tar.gz: c6f4d4e13582c15045bc040815f88c5d07562ecbb1554daccd6ea6383462ff7c56f3c0adeecc8c4eb29f800faf865a896f0edba4a80979f35c5d46fd897ba72d

data/lib/irt_ruby/rasch_model.rb

@@ -0,0 +1,59 @@
+# frozen_string_literal: true
+
+require "matrix"
+
+module IrtRuby
+  # A class representing the Rasch model for Item Response Theory.
+  class RaschModel
+    def initialize(data, max_iter: 1000, tolerance: 1e-6, learning_rate: 0.01)
+      @data = data
+      @abilities = Array.new(data.row_count) { rand }
+      @difficulties = Array.new(data.column_count) { rand }
+      @max_iter = max_iter
+      @tolerance = tolerance
+      @learning_rate = learning_rate
+    end
+
+    # Sigmoid function to calculate probability
+    def sigmoid(x)
+      1.0 / (1.0 + Math.exp(-x))
+    end
+
+    # Calculate the log-likelihood of the data given the current parameters
+    def likelihood
+      likelihood = 0
+      @data.row_vectors.each_with_index do |row, i|
+        row.to_a.each_with_index do |response, j|
+          prob = sigmoid(@abilities[i] - @difficulties[j])
+          likelihood += response == 1 ? Math.log(prob) : Math.log(1 - prob)
+        end
+      end
+      likelihood
+    end
+
+    # Update parameters using gradient ascent
+    def update_parameters
+      last_likelihood = likelihood
+      @max_iter.times do |_iter|
+        @data.row_vectors.each_with_index do |row, i|
+          row.to_a.each_with_index do |response, j|
+            prob = sigmoid(@abilities[i] - @difficulties[j])
+            error = response - prob
+            @abilities[i] += @learning_rate * error
+            @difficulties[j] -= @learning_rate * error
+          end
+        end
+        current_likelihood = likelihood
+        break if (last_likelihood - current_likelihood).abs < @tolerance
+
+        last_likelihood = current_likelihood
+      end
+    end
+
+    # Fit the model to the data
+    def fit
+      update_parameters
+      { abilities: @abilities, difficulties: @difficulties }
+    end
+  end
+end

data/lib/irt_ruby/three_parameter_model.rb

@@ -0,0 +1,74 @@
+# frozen_string_literal: true
+
+require "matrix"
+
+module IrtRuby
+  # A class representing the Three-Parameter model for Item Response Theory.
+  class ThreeParameterModel
+    def initialize(data, max_iter: 1000, tolerance: 1e-6, learning_rate: 0.01)
+      @data = data
+      @abilities = Array.new(data.row_count) { rand }
+      @difficulties = Array.new(data.column_count) { rand }
+      @discriminations = Array.new(data.column_count) { rand }
+      @guessings = Array.new(data.column_count) { rand * 0.3 }
+      @max_iter = max_iter
+      @tolerance = tolerance
+      @learning_rate = learning_rate
+    end
+
+    # Sigmoid function to calculate probability
+    def sigmoid(x)
+      1.0 / (1.0 + Math.exp(-x))
+    end
+
+    # Probability function for the 3PL model
+    def probability(theta, a, b, c)
+      c + (1 - c) * sigmoid(a * (theta - b))
+    end
+
+    # Calculate the log-likelihood of the data given the current parameters
+    def likelihood
+      likelihood = 0
+      @data.row_vectors.each_with_index do |row, i|
+        row.to_a.each_with_index do |response, j|
+          prob = probability(@abilities[i], @discriminations[j], @difficulties[j], @guessings[j])
+          likelihood += response == 1 ? Math.log(prob) : Math.log(1 - prob)
+        end
+      end
+      likelihood
+    end
+
+    # Update parameters using gradient ascent
+    def update_parameters
+      last_likelihood = likelihood
+      @max_iter.times do |_iter|
+        @data.row_vectors.each_with_index do |row, i|
+          row.to_a.each_with_index do |response, j|
+            prob = probability(@abilities[i], @discriminations[j], @difficulties[j], @guessings[j])
+            error = response - prob
+            @abilities[i] += @learning_rate * error * @discriminations[j]
+            @difficulties[j] -= @learning_rate * error * @discriminations[j]
+            @discriminations[j] += @learning_rate * error * (@abilities[i] - @difficulties[j])
+            @guessings[j] += @learning_rate * error * (1 - prob)
+            @guessings[j] = [[@guessings[j], 0].max, 1].min # Keep guessings within [0, 1]
+          end
+        end
+        current_likelihood = likelihood
+        break if (last_likelihood - current_likelihood).abs < @tolerance
+
+        last_likelihood = current_likelihood
+      end
+    end
+
+    # Fit the model to the data
+    def fit
+      update_parameters
+      {
+        abilities: @abilities,
+        difficulties: @difficulties,
+        discriminations: @discriminations,
+        guessings: @guessings
+      }
+    end
+  end
+end

data/lib/irt_ruby/two_parameter_model.rb

@@ -0,0 +1,65 @@
+# frozen_string_literal: true
+
+require "matrix"
+
+module IrtRuby
+  # A class representing the Two-Parameter model for Item Response Theory.
+  class TwoParameterModel
+    def initialize(data, max_iter: 1000, tolerance: 1e-6, learning_rate: 0.01)
+      @data = data
+      @abilities = Array.new(data.row_count) { rand }
+      @difficulties = Array.new(data.column_count) { rand }
+      @discriminations = Array.new(data.column_count) { rand }
+      @max_iter = max_iter
+      @tolerance = tolerance
+      @learning_rate = learning_rate
+    end
+
+    # Sigmoid function
+    def sigmoid(x)
+      1.0 / (1.0 + Math.exp(-x))
+    end
+
+    # Calculate the log-likelihood of the data given the current parameters
+    def likelihood
+      likelihood = 0
+      @data.row_vectors.each_with_index do |row, i|
+        row.to_a.each_with_index do |response, j|
+          prob = sigmoid(@discriminations[j] * (@abilities[i] - @difficulties[j]))
+          if response == 1
+            likelihood += Math.log(prob)
+          elsif response.zero?
+            likelihood += Math.log(1 - prob)
+          end
+        end
+      end
+      likelihood
+    end
+
+    # Update parameters using gradient ascent
+    def update_parameters
+      last_likelihood = likelihood
+      @max_iter.times do |_iter|
+        @data.row_vectors.each_with_index do |row, i|
+          row.to_a.each_with_index do |response, j|
+            prob = sigmoid(@discriminations[j] * (@abilities[i] - @difficulties[j]))
+            error = response - prob
+            @abilities[i] += @learning_rate * error * @discriminations[j]
+            @difficulties[j] -= @learning_rate * error * @discriminations[j]
+            @discriminations[j] += @learning_rate * error * (@abilities[i] - @difficulties[j])
+          end
+        end
+        current_likelihood = likelihood
+        break if (last_likelihood - current_likelihood).abs < @tolerance
+
+        last_likelihood = current_likelihood
+      end
+    end
+
+    # Fit the model to the data
+    def fit
+      update_parameters
+      { abilities: @abilities, difficulties: @difficulties, discriminations: @discriminations }
+    end
+  end
+end

data/lib/irt_ruby/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module IrtRuby
+  VERSION = "0.1.0"
+end

data/lib/irt_ruby.rb

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+require "irt_ruby/version"
+require "irt_ruby/rasch_model"
+require "irt_ruby/two_parameter_model"
+require "irt_ruby/three_parameter_model"
+
+module IrtRuby
+  class Error < StandardError; end
+end

metadata

@@ -0,0 +1,97 @@
+--- !ruby/object:Gem::Specification
+name: irt_ruby
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Alex Kholodniak
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2024-06-09 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+description: IrtRuby is a Ruby gem that provides implementations of the Rasch model,
+  Two-Parameter model, and Three-Parameter model for Item Response Theory (IRT). It
+  allows you to estimate the abilities of individuals and the difficulties, discriminations,
+  and guessing parameters of items based on their responses to a set of items.
+email:
+- alexandrkholodniak@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/irt_ruby.rb
+- lib/irt_ruby/rasch_model.rb
+- lib/irt_ruby/three_parameter_model.rb
+- lib/irt_ruby/two_parameter_model.rb
+- lib/irt_ruby/version.rb
+homepage: https://github.com/SyntaxSpirits/irt_ruby
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/SyntaxSpirits/irt_ruby
+  source_code_uri: https://github.com/SyntaxSpirits/irt_ruby
+  changelog_uri: https://github.com/SyntaxSpirits/irt_ruby/CHANGELOG.md
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '2.6'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.4.16
+signing_key:
+specification_version: 4
+summary: A Ruby gem that provides implementations of Rasch, Two-Parameter, and Three-Parameter
+  models for Item Response Theory (IRT).
+test_files: []