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