cmd_stan_rb 0.2.0 → 0.3.0

data/lib/cmd_stan_rb/version.rb

@@ -1,3 +1,3 @@
 module CmdStanRb
-  VERSION = "0.2.0"
+  VERSION = "0.3.0"
 end

data/lib/stan.rb

@@ -1,5 +1,7 @@
 require "stan/model"
 require "stan/examples"
+require "stan/histogram"
+require "stan/fit_result"
 
 module Stan
 end

data/lib/stan/fit_result.rb

@@ -0,0 +1,31 @@
+module Stan
+  class FitResult
+    attr_reader :rows, :headers, :data_rows
+
+    def initialize(original_output)
+      @rows = Array(original_output.split("\n"))
+      @headers = rows[38].to_s.split(",") # In L39 there is the header column
+      @data_rows = rows[43..].map { |r| r.split(",") } # From L44 on there is the raw data!
+
+      define_histograms!
+    end
+
+    def parameters
+      headers[7..]
+    end
+
+    private
+
+    def define_histograms!
+      headers.each do |parameter|
+        define_singleton_method parameter do
+          parameter_index = headers.index(parameter)
+          data_rows.each_with_object({}) do |row, hash|
+            v = row[parameter_index].to_f.round(1)
+            hash[v] = (hash[v] || 0) + 1
+          end.sort.to_h
+        end
+      end
+    end
+  end
+end

data/lib/stan/histogram.rb

@@ -0,0 +1,27 @@
+module Stan
+  class Histogram
+    attr_reader :list, :bin_size
+
+    def initialize(list, bin_size: nil)
+      @list = list
+      @bin_size = bin_size
+    end
+
+    def to_a
+      @list.each_with_object({}) do |item, hash|
+        bin =
+          if bin_size.to_f > 0
+            (item.to_f / bin_size).round * bin_size
+          else
+            item
+          end
+
+        hash[bin] = hash[bin].to_i + 1
+      end.sort
+    end
+
+    def to_h
+      to_a.to_h
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: cmd_stan_rb
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - 'Robin Neumann
@@ -10,8 +10,22 @@ authors:
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-05-10 00:00:00.000000000 Z
-dependencies: []
+date: 2020-05-29 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Stan is a state-of-the-art platform for statistical modeling and high-performance
   statistical computation. Thousands of users rely on Stan for statistical modeling,
   data analysis, and prediction in the social, biological, and physical sciences,
@@ -36,11 +50,15 @@ files:
 - bin/setup
 - cmd_stan_rb.gemspec
 - demo.ipynb
+- examples/exponential.ipynb
+- examples/linear_regression.ipynb
 - lib/cmd_stan_rb.rb
 - lib/cmd_stan_rb/configuration.rb
 - lib/cmd_stan_rb/version.rb
 - lib/stan.rb
 - lib/stan/examples.rb
+- lib/stan/fit_result.rb
+- lib/stan/histogram.rb
 - lib/stan/model.rb
 homepage: https://github.com/neumanrq/cmd_stan_rb
 licenses: []