Solving Inconsistent Number of Samples Error in Train-Test Split Process for Machine Learning