summaryrefslogtreecommitdiff
path: root/cv/data/utils/count_features.rb
blob: c8a5893ec9a1a7d7a7213e1dc4b054d301658955 (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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
require 'rubygems'
require 'opentox-ruby'
require 'yaml'

@subjectid = nil

if ARGV.size != 1
  puts "Args: path/to/dataset.yaml"
  puts ARGV.size
  exit
end

def count_features(ds_uri)
  puts ds_uri

  dataset = OpenTox::Dataset.find(ds_uri, @subjectid)

  features = dataset.features.keys
  puts "# all features: #{features.size}"
  feature_names = []
  delete_features = []
  features.each{ |fn|
    feature_names << fn.split("\/feature\/").last
    dataset.features[fn][RDF.type].each { |typestr|
      if typestr.include? "MissingFeature"
        delete_features << fn 
        @missing_features << dataset.features[fn][DC.title]
      end
    }
  }
  @all_feature_names << feature_names.sort
  @all_feature_names << "" 

  puts "# Missingfeatures: #{delete_features.size}"
  features = features - delete_features
  puts "# numeric features: #{features.size}"
  puts "-----"
end


@missing_features = []
@all_feature_names = []
path = ARGV[0]
puts path
ds = YAML::load_file("#{path}")
#ds = YAML::load_file("../datasets.yaml")
ds.keys.each { |dataset|
  puts "----------"
  puts dataset
  @all_feature_names << "" 
  @all_feature_names << "------ new dataset ------" 
  @all_feature_names << "-------- #{dataset} --------"
  ds[dataset].keys.each { |pc|
    if !(pc == "dataset") || (pc == "test") || (pc == "training")
      puts pc
      @all_feature_names << "--- new feature: #{pc} ---" 
      count_features(ds[dataset][pc])
    end
  }
  puts "----------"
  puts
}
puts
puts "Missing features over all datasets:"
puts @missing_features.uniq!.to_yaml
puts 
puts "All feature names:"
puts @all_feature_names