Skip to content
Snippets Groups Projects
Commit 57b2a4b1 authored by lukas leufen's avatar lukas leufen
Browse files

Merge branch 'lukas_issue070_bug_wrong-model-plot-path' into 'master'

Resolve "BUG: wrong model plot path"

Closes #70

See merge request toar/machinelearningtools!64
parents 3725c58e 7997cf3c
No related branches found
No related tags found
2 merge requests!67apply bug fix to dev too,!64Resolve "BUG: wrong model plot path"
Pipeline #31544 passed
......@@ -100,5 +100,5 @@ class ModelSetup(RunEnvironment):
def plot_model(self): # pragma: no cover
with tf.device("/cpu:0"):
file_name = f"{self.model_name.split(sep='.')[0]}.pdf"
file_name = f"{self.model_name.rsplit('.', 1)[0]}.pdf"
keras.utils.plot_model(self.model, to_file=file_name, show_shapes=True, show_layer_names=True)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment