use png format for plots avoiding css overwrite from svg objects and better browser performance

1 files changed, 13 insertions, 0 deletions

diff --git a/application.rb b/application.rb
index 0d5da57..f5f3486 100644
--- a/application.rb
+++ b/application.rb
@@ -102,6 +102,19 @@ get '/predict/modeldetails/:model' do
   training_dataset = model.model.training_dataset
   data_entries = training_dataset.data_entries
   crossvalidations = model.crossvalidations
+  if model.classification?
+    crossvalidations.each do |cv|
+      File.open(File.join('public', "#{cv.id}.png"), 'w') do |file|
+        file.write(cv.probability_plot(format: "png"))
+      end unless File.exists? File.join('public', "#{cv.id}.png")
+    end
+  else
+    crossvalidations.each do |cv|
+      File.open(File.join('public', "#{cv.id}.png"), 'w') do |file|
+        file.write(cv.correlation_plot(format: "png"))
+      end unless File.exists? File.join('public', "#{cv.id}.png")
+    end
+  end
 
   response['Content-Type'] = "text/html"
   return haml :model_details, :layout=> false, :locals => {:model => model,