diff options
Diffstat (limited to 'scripts/crossvalidation.rb')
| -rwxr-xr-x | scripts/crossvalidation.rb | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/scripts/crossvalidation.rb b/scripts/crossvalidation.rb index 9657af1..c1653dc 100755 --- a/scripts/crossvalidation.rb +++ b/scripts/crossvalidation.rb @@ -5,9 +5,8 @@ require 'yaml' name = File.basename ARGV[0], ".csv" file = File.join "data",ARGV[0] dataset = Dataset.from_csv_file file -model = Model::LazarRegression.create(training_dataset: dataset)#, :prediction_algorithm => "OpenTox::Algorithm::Regression.local_fingerprint_regression") -#model = Model::LazarRegression.create(dataset, :prediction_algorithm => "OpenTox::Algorithm::Regression.local_physchem_regression") -#model = Model::LazarRegression.create(dataset, :prediction_algorithm => "OpenTox::Algorithm::Regression.local_weighted_average") +#model = Model::LazarRegression.create(training_dataset: dataset)#, :prediction_algorithm => "OpenTox::Algorithm::Regression.local_fingerprint_regression") +model = Model::LazarRegression.create(training_dataset: dataset, algorithms: { :similarity => { :min => 0.3 }}) csv_file = File.join("data",ARGV[0].sub(/.csv/,"-cv-#{ARGV[1]}.csv")) id_file = File.join("data",ARGV[0].sub(/.csv/,"-cv-#{ARGV[1]}.id")) cv = Validation::RegressionCrossValidation.create model |