correct = predicted.collect.with_index do |e,i|
              if (e.to_f - validation_y_data[i].to_f).magnitude <= @magnitude
                1
              else
                0