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
|
require_relative "setup.rb"
class LazarExtendedTest < MiniTest::Test
def test_lazar_bbrc_ham_minfreq
dataset = OpenTox::MeasuredDataset.new
dataset.upload File.join(DATA_DIR,"hamster_carcinogenicity.csv")
model = OpenTox::Model::Lazar.create OpenTox::Algorithm::Fminer.bbrc(:dataset => dataset, :min_frequency => 5)
feature_dataset = OpenTox::CalculatedDataset.find model.feature_dataset_id
assert_equal dataset.compounds.size, feature_dataset.compounds.size
assert_equal 41, feature_dataset.features.size
assert_equal '[#7&A]-[#6&A]=[#7&A]', feature_dataset.features.first.title
compound = OpenTox::Compound.from_inchi("InChI=1S/C6H6/c1-2-4-6-5-3-1/h1-6H")
prediction_dataset = model.predict :compound => compound
prediction = prediction_dataset.data_entries.first
assert_equal "false", prediction.first
assert_equal 0.12380952380952381, prediction.last
dataset.delete
model.delete
feature_dataset.delete
prediction_dataset.delete
end
def test_lazar_bbrc_large_ds
# TODO fminer crashes with these settings
dataset = OpenTox::MeasuredDataset.new
dataset.upload File.join(DATA_DIR,"multi_cell_call_no_dup.csv")
feature_dataset = OpenTox::Algorithm::Fminer.bbrc(:dataset => dataset)#, :min_frequency => 15)
model = OpenTox::Model::Lazar.create feature_dataset
model.save
p model.id
feature_dataset = OpenTox::CalculatedDataset.find model.feature_dataset_id
assert_equal dataset.compounds.size, feature_dataset.compounds.size
assert_equal 52, feature_dataset.features.size
assert_equal '[#17&A]-[#6&A]', feature_dataset.features.first.title
compound = OpenTox::Compound.from_inchi("InChI=1S/C10H9NO2S/c1-8-2-4-9(5-3-8)13-6-10(12)11-7-14/h2-5H,6H2,1H3")
prediction_dataset = model.predict :compound => compound
prediction = prediction_dataset.data_entries.first
assert_in_delta 0.025, prediction[:confidence], 0.001
#assert_equal 0.025885845574483608, prediction[:confidence]
# with compound change in training_dataset see:
# https://github.com/opentox/opentox-test/commit/0e78c9c59d087adbd4cc58bab60fb29cbe0c1da0
#assert_equal 0.02422364949075546, prediction[:confidence]
dataset.delete
model.delete
feature_dataset.delete
prediction_dataset.delete
end
def test_lazar_kazius
# TODO find a solution for feature datasets > 16M (size limit in mongodb)
dataset = OpenTox::MeasuredDataset.from_csv_file File.join(DATA_DIR,"kazius.csv")
feature_dataset = OpenTox::Algorithm::Fminer.bbrc(:dataset => dataset, :min_frequency => 100)
assert_equal feature_dataset.compounds.size, dataset.compounds.size
model = OpenTox::Model::Lazar.create feature_dataset
p model.id
dataset.delete
#feature_dataset.delete
end
end
|