diff options
Diffstat (limited to 'report/plot_factory.rb')
-rw-r--r-- | report/plot_factory.rb | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/report/plot_factory.rb b/report/plot_factory.rb index a2be092..093e96a 100644 --- a/report/plot_factory.rb +++ b/report/plot_factory.rb @@ -123,8 +123,9 @@ module Reports Reports::r_util.double_hist_plot(out_files, train, test, numerical, numerical, "Training Data", "Test Data", "Prediction Feature Distribution", validation_set.validations.first.prediction_feature ) else + binary = validation_set.validations[0].bbrc_training_features Reports::r_util.feature_value_plot(out_files, validation_set.validations[0].training_feature_dataset_uri, - validation_set.validations[0].test_feature_dataset_uri, "Training Data", "Test Data", + validation_set.validations[0].test_feature_dataset_uri, "Training Data", "Test Data", (binary ? "binary" : "numerical"), validation_set.validations[0].prediction_feature, validation_set.validations[0].subjectid, waiting_task ) end end |