NumberOfRowsTests = [100, 500, 1000, 2000, 8000, 32561] for num_rows in (NumberOfRowsTests): model = sklearn.naive_bayes.MultinomialNB() model.fit(income_feature_vecs[:num_rows, :], income_labels[:num_rows])