Commit dd006670 authored by Jean Ibarz's avatar Jean Ibarz
Browse files

Update train_model.py

parent ba8c39c8
......@@ -170,12 +170,12 @@ if __name__ == '__main__':
signals=np.array(training_df['signal']),
labels=np.array(training_df['azimuth']),
sounds=training_sounds,
n_samples=n_samples,
n_samples=None,
n_augment=1)
test_signals, test_labels = generate_augmented_labelled_dataset(signals=np.array(test_df['signal']),
labels=np.array(test_df['azimuth']),
sounds=test_sounds,
n_samples=n_samples,
n_samples=None,
n_augment=1)
if exp_config['model_name'] == 'left_center_right':
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment