From d25f286e76edca37b5ca6d4a1b815cfdaa2b74cc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?K=C3=BChle=2C=20Laura=20Christine=20=28lakue103=29?=
<laura.kuehle@uni-duesseldorf.de>
Date: Wed, 24 Nov 2021 17:51:20 +0100
Subject: [PATCH] Improved directory selection.
---
ANN_Data_Generator.py | 7 +++----
1 file changed, 3 insertions(+), 4 deletions(-)
diff --git a/ANN_Data_Generator.py b/ANN_Data_Generator.py
index 07a7d7c..e9a9b8c 100644
--- a/ANN_Data_Generator.py
+++ b/ANN_Data_Generator.py
@@ -16,6 +16,7 @@ TODO: Fix bug in 'input_data' calculation -> Done
TODO: Change order of imports -> Done
TODO: Improve calculation of normalization -> Done
TODO: Improve calculation of random point for stencil -> Done
+TODO: Improve directory selection -> Done
"""
@@ -47,7 +48,7 @@ class TrainingDataGenerator(object):
"""
def __init__(self, initial_conditions, left_bound=-1, right_bound=1, balance=0.5,
- stencil_length=3, directory=None):
+ stencil_length=3, directory='test_data'):
"""Initializes TrainingDataGenerator.
Parameters
@@ -85,9 +86,7 @@ class TrainingDataGenerator(object):
self._troubled_functions.append(function)
# Set directory
- self._data_dir = 'test_data'
- if directory is not None:
- self._data_dir = directory
+ self._data_dir = directory
if not os.path.exists(self._data_dir):
os.makedirs(self._data_dir)
--
GitLab