Skip to content
Snippets Groups Projects
Commit ec9b3979 authored by Benjamin Paul Jaeger's avatar Benjamin Paul Jaeger
Browse files

return heatmap result from plot function

parent f46cfe93
No related branches found
No related tags found
1 merge request!2Add Z0 callback
Pipeline #4095766 failed
Stage: test
    Stage: build
      Hide whitespace changes
      Inline Side-by-side
      Showing
      with 2 additions and 2 deletions
      freeforestml/plot.py
      + 2
      − 2
      View file @ ec9b3979
      ...@@ -579,7 +579,7 @@ def correlation_matrix(df, variables, ...@@ -579,7 +579,7 @@ def correlation_matrix(df, variables,
      columns=[v.name for v in variables], columns=[v.name for v in variables],
      index=[v.name for v in variables]) index=[v.name for v in variables])
      sns.heatmap(data * 100, **dict(vmin=-100, vmax=100, cmap="RdBu_r", ax=axes, res = sns.heatmap(data * 100, **dict(vmin=-100, vmax=100, cmap="RdBu_r", ax=axes,
      cbar_kws={ cbar_kws={
      'label': "Correlation coefficient $\\rho \\cdot 100$" 'label': "Correlation coefficient $\\rho \\cdot 100$"
      } }
      ...@@ -591,4 +591,4 @@ def correlation_matrix(df, variables, ...@@ -591,4 +591,4 @@ def correlation_matrix(df, variables,
      atlasify(*fill_labels(atlas, info), axes=axes, outside=True) atlasify(*fill_labels(atlas, info), axes=axes, outside=True)
      figure.subplots_adjust(top=1/enlarge) figure.subplots_adjust(top=1/enlarge)
      return data return res, data
      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