03_classifier_training_sample_generator.py

from Experiment import Experiment
from Vectors import Vectors
from TrainingSampleGenerator import TrainingSampleGenerator
from ClassifierTrainer import ClassifierTrainer
from list_of_experiments import LIST_OF_EXPERIMENTS


def main():
    for experiment_name in LIST_OF_EXPERIMENTS:
        experiment = Experiment(experiment_name)

        vectors = Vectors(experiment)

        sample_generator = TrainingSampleGenerator(experiment, vectors)
        sample_generator.generate_samples()

        print(experiment.variables["vector_tag"])
        classifier = ClassifierTrainer(experiment)
        classifier.load_samples(sample_generator)
        classifier.generate_result()
        classifier.save_result()


if __name__ == '__main__':
    main()