I\'m trying to practice this code for ML, but I\'m facing some error saying "ValueError: Found input variables with inconsistent numbers of samples: [2788, 1813]"