Commit 57b2a4b1 authored by lukas leufen's avatar lukas leufen

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
Pipeline #31544 passed with stages
in 7 minutes and 27 seconds
......@@ -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)
Markdown is supported
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