def test_visualize_mse_bars(self):
filename = '../../data/emotiv/EEG_Data_filtered.csv'
dataset = DataReader.read_data(filename, ',')
training_set, test_set = ExperimentorService.split_dataset(dataset, ratio=0.2)
artifact_size = 20
artifact_dataset, _ = ExperimentorService.artifactify(test_set, artifact_size, randomly_add_artifacts=True)
window_sizes = range(10, 301, 5)
Visualizer.visualize_cross_validation_bars(training_set, test_set, artifact_dataset, window_sizes, name="figure_cross_validation_bars2")
def test_visualize_mse_bars_no_artifacts(self):
filename = '../../data/emotiv/EEG_Data_filtered.csv'
dataset = DataReader.read_data(filename, ',')
training_set, test_set = ExperimentorService.split_dataset(dataset, ratio=0.2)
window_size = 40
threshold_max, threshold_avg, threshold_avg_max = ExperimentorService.calibrate(training_set, window_size)
thresholds = [threshold_max, threshold_avg, threshold_avg_max]
window_sizes = [10, 15, 20, 25, 30, 35, 40, 45, 50, 55, 60, 65, 70, 75, 80, 85, 90, 95, 100]
Visualizer.visualize_cross_validation_bars(test_set, test_set, thresholds, window_sizes, name="figure_cross_validation_bars_no_artifacts")
