diff options
Diffstat (limited to 'lib/transform.rb')
-rw-r--r-- | lib/transform.rb | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/lib/transform.rb b/lib/transform.rb index cbfa915..fad9517 100644 --- a/lib/transform.rb +++ b/lib/transform.rb @@ -400,10 +400,11 @@ module OpenTox # Converts fingerprints to matrix, order of rows by fingerprints. nil values allowed. # Same for compound fingerprints. def get_matrices - @compounds = @model.training_compounds.clone - @activities = @model.training_activities.clone - @n_prop = @model.training_fingerprints.clone - @q_prop = @model.query_fingerprint.clone + @compounds = @model.training_dataset.compounds.clone + # TODO select predicted variable + @activities = @model.training_activities + @n_prop = @model.feature_dataset.data_entries.clone + @q_prop = @model.query_fingerprint.flatten.clone end # Returns propositionalized data, if appropriate, or nil |