summaryrefslogtreecommitdiff
path: root/lib/feature.rb
blob: 0ca4d41b44025caeb735e7b4d977fdc95f9ad896 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
module OpenTox

  # Basic feature class
  class Feature
    field :measured, type: Boolean
    field :calculated, type: Boolean
    field :category, type: String
    field :unit, type: String
    field :conditions, type: Hash

    def nominal?
      self.class == NominalFeature
    end

    def numeric?
      self.class == NumericFeature
    end
  end

  # Feature for categorical variables
  class NominalFeature < Feature
    field :accept_values, type: Array
  end

  # Feature for quantitative variables
  class NumericFeature < Feature
  end

  # Feature for SMARTS fragments
  class Smarts < NominalFeature
    field :smarts, type: String 
    index "smarts" => 1
    def self.from_smarts smarts
      self.find_or_create_by :smarts => smarts
    end
  end

end