|
3 | 3 | import sys |
4 | 4 | stderr = sys.stderr |
5 | 5 | sys.stderr = open(os.devnull, 'w') |
6 | | -import keras |
7 | | -sys.stderr = stderr |
8 | | -# to use CPU uncomment below code |
9 | | -os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" # see issue #152 |
10 | | -os.environ["CUDA_VISIBLE_DEVICES"] = "-1" |
11 | | -# disable tensorflow logs |
12 | | -os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' |
13 | 6 | import tensorflow as tf |
14 | 7 |
|
15 | | -config = tf.ConfigProto(intra_op_parallelism_threads=5, |
16 | | - inter_op_parallelism_threads=5, |
17 | | - allow_soft_placement=True, |
18 | | - device_count = {'CPU' : 1, |
19 | | - 'GPU' : 0} |
20 | | - ) |
21 | | -from keras.layers import LSTM, GRU, Dense, Activation, LeakyReLU, Dropout |
22 | | -from keras.layers import Conv1D, MaxPool1D, GlobalAveragePooling1D |
23 | | -from keras.models import Sequential |
24 | | -from keras.callbacks import ModelCheckpoint, TensorBoard |
25 | | -from keras.utils import to_categorical |
| 8 | +from tensorflow.keras.layers import LSTM, GRU, Dense, Activation, LeakyReLU, Dropout |
| 9 | +from tensorflow.keras.layers import Conv1D, MaxPool1D, GlobalAveragePooling1D |
| 10 | +from tensorflow.keras.models import Sequential |
| 11 | +from tensorflow.keras.callbacks import ModelCheckpoint, TensorBoard |
| 12 | +from tensorflow.keras.utils import to_categorical |
26 | 13 |
|
27 | 14 | from sklearn.metrics import accuracy_score, mean_absolute_error, confusion_matrix |
28 | 15 |
|
@@ -264,7 +251,7 @@ def train(self, override=False): |
264 | 251 | model_filename = self._get_model_filename() |
265 | 252 |
|
266 | 253 | self.checkpointer = ModelCheckpoint(model_filename, save_best_only=True, verbose=1) |
267 | | - self.tensorboard = TensorBoard(log_dir=f"logs/{self.model_name}") |
| 254 | + self.tensorboard = TensorBoard(log_dir=os.path.join("logs", self.model_name)) |
268 | 255 |
|
269 | 256 | self.history = self.model.fit(self.X_train, self.y_train, |
270 | 257 | batch_size=self.batch_size, |
|
0 commit comments