#print(plotter.prep.selected_words)

        elif b2 == 3:

            start = time.time()

            methods = ['GMM', 'MEAN', 'RandomForest', 'SVM', 'KNN']

            for method in methods:
                print("Running " + method)
                getattr(plotter, method)(show=True)

            end = time.time()
            print(end - start, " seconds")

        elif b2 == 4:

            start = time.time()

            plotter.run_all_kfold()

            end = time.time()

            print(end - start, " seconds")

        b2 = int(
            input(
                '\nChoose action:\n1)Visualize Bigrams\n2)Visualize Terms\n3)Run model grid param search\n4)Run all '
                'models kfold\n5)Exit\n> '))
        assert b2 in range(1, 5)