data_modeler 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 3daf6a98719d10349658f22bda03ae9763dfe2fd
-  data.tar.gz: e7ce8d15928b20dbfda50ea2fafc910bb4f43451
+  metadata.gz: 7be4c7a2b2fcc56f06898a1fa67fc3494d8560c5
+  data.tar.gz: 27c3a704d5827dc86cac1f27652b178732fa8451
 SHA512:
-  metadata.gz: 8a8135391faa041a1ac876c77f06add9311a9c58b0a300d774b28cff0bb9a7ce6e27d11e7bdcf1f085e545f755b82580b38eecf0a898ee29027c376abc95c289
-  data.tar.gz: dfaad0abc476b8cee5b357a319be2427b97fe6e69ca10cd7a304ac7993b65df936801ea4df78860478780c9946f42a57629515d40d102bdceaf01161e472fe50
+  metadata.gz: e7a0847efa67279ca0aeb8ce808ed8cc4aef8c89e430fb857fd766524e66615a06548907a578982c8cc4f2fdab4ac1764d7af9674ca7667ff7945b81eb7fcc50
+  data.tar.gz: 2c5701cb6d89e66258baf0ef8883cb63a02a1c21dccc2cc810d737f9802ef3b725980e02a6982f057e4bad9805fe00ac1dbc73db97222ad152dd5eb550a59e6c

data/README.md

@@ -1,4 +1,9 @@
-# DataModeler
+
+# [Data Modeler](https://github.com/giuse/data_modeler)
+
+[![Gem Version](https://badge.fury.io/rb/data_modeler.svg)](https://badge.fury.io/rb/data_modeler)
+[![Build Status](https://travis-ci.org/giuse/data_modeler.svg?branch=master)](https://travis-ci.org/giuse/data_modeler)
+[![Code Climate](https://codeclimate.com/github/giuse/data_modeler/badges/gpa.svg)](https://codeclimate.com/github/giuse/data_modeler)
 
 Using machine learning, create generative models based on your data alone. Applications span from imputation to prediction.
 

data/data_modeler.gemspec

@@ -21,6 +21,10 @@ Gem::Specification.new do |spec|
   spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
   spec.require_paths = ["lib"]
 
+  # Models
+  # TODO: learn to keep them independent from the gem (plug-ins)
+  spec.add_dependency 'ruby-fann', '~>1.2.6'
+
   # Debug
   spec.add_development_dependency 'pry', '~> 0.10'
   spec.add_development_dependency 'pry-nav', '~> 0.2'
@@ -30,5 +34,6 @@ Gem::Specification.new do |spec|
   # Test
   spec.add_development_dependency 'bundler', '~> 1.14'
   spec.add_development_dependency 'rake', '~> 10.0'
-  spec.add_development_dependency 'rspec', '~> 3.0'
+  spec.add_development_dependency 'rspec', '~> 3.6'
+  spec.add_development_dependency 'rspec-retry', '~> 0.5.4'
 end

data/lib/data_modeler/model/fann.rb

@@ -0,0 +1,52 @@
+require 'ruby-fann'
+
+# Model class based on Fast Artificial Neural Networks (FANN)
+class DataModeler::Model::FANN
+
+  attr_reader :opts, :fann, :algo, :actfn
+
+  def initialize netstruct, algo: nil, actfn: nil
+    ninputs, hidden_layers, noutputs = netstruct
+    @opts = {
+      num_inputs: ninputs,
+      hidden_neurons: hidden_layers,
+      num_outputs: noutputs
+    }
+    @algo = algo
+    @actfn = actfn
+  end
+
+  def reset
+    @fann = RubyFann::Standard.new opts
+
+    if algo
+      fann.set_training_algorithm(algo)
+    end
+
+    if actfn
+      fann.set_activation_function_hidden(actfn)
+      fann.set_activation_function_output(actfn)
+    end
+
+    return self # allows chaining for `model.reset.train`
+  end
+
+  def train ngens, trainset
+    tset = RubyFann::TrainData.new(
+      inputs: trainset[:input], desired_outputs: trainset[:target])
+    # fann.init_weights tset # test this weights initialization
+
+    # params: train_data, max_epochs, reports_interval, desired_error
+    fann.train_on_data(tset, ngens, 1000, 1e-10)
+  end
+
+  def test inputs
+    inputs.collect { |i| fann.run i }
+  end
+
+  def save filename
+    # can do filename check here...?
+    # I'd rather have kind of a `to_s`, and do the saving in the modeler
+    fann.save filename
+  end
+end

data/lib/data_modeler/version.rb

@@ -1,5 +1,5 @@
 # Main gem module
 module DataModeler
   # Version number
-  VERSION = "0.1.1"
+  VERSION = "0.2.0"
 end

data/lib/data_modeler.rb

@@ -1,5 +1,11 @@
 require "data_modeler/version"
 require "data_modeler/exceptions"
-require "data_modeler/dataset_helper"
-require "data_modeler/dataset"
-require "data_modeler/dataset_gen"
+
+# Dataset
+require "data_modeler/dataset/dataset_helper"
+require "data_modeler/dataset/dataset"
+require "data_modeler/dataset/dataset_gen"
+
+# Models (should be added to this module)
+module DataModeler::Model; end
+require "data_modeler/model/fann"

metadata

@@ -1,15 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: data_modeler
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - Giuseppe Cuccu
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-05-12 00:00:00.000000000 Z
+date: 2017-05-13 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: ruby-fann
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.2.6
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.2.6
 - !ruby/object:Gem::Dependency
   name: pry
   requirement: !ruby/object:Gem::Requirement
@@ -100,14 +114,28 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.0'
+        version: '3.6'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.6'
+- !ruby/object:Gem::Dependency
+  name: rspec-retry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.5.4
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.0'
+        version: 0.5.4
 description: Using machine learning, create generative models based on your data alone.
   Applications span from imputation to prediction. This build specifically leverages
   time series. Further work on data preparation will be released as a separate project.
@@ -128,10 +156,11 @@ files:
 - bin/setup
 - data_modeler.gemspec
 - lib/data_modeler.rb
-- lib/data_modeler/dataset.rb
-- lib/data_modeler/dataset_gen.rb
-- lib/data_modeler/dataset_helper.rb
+- lib/data_modeler/dataset/dataset.rb
+- lib/data_modeler/dataset/dataset_gen.rb
+- lib/data_modeler/dataset/dataset_helper.rb
 - lib/data_modeler/exceptions.rb
+- lib/data_modeler/model/fann.rb
 - lib/data_modeler/version.rb
 homepage: https://github.com/giuse/data_modeler
 licenses: