Include optional colormaps in "nolearn/lasagne/visualize.py" ?

[jorgdv]

Hi there,

I think that the function visualize.py should include as input an optional colormap parameter, in order to see the plots as each one likes. Now the colormap "gray" is set in the function, so you can't even change it outside.

Best regards

[BenjaminBossan]

The functions in visualize.py return the plot object. I don't know whether this works, but maybe it's possible to override some attributes and re-draw the images.

If this does not work, it would probably best to be able to pass arbitrary kwargs, since others might want to change other parameters.

[jorgdv]

That was my point. I think it is so much easier to just pass arbitrary arguments, since the output (plot object as you said) is already drawn with gray color map, and I don't think it is possible to override the atributes.

[BenjaminBossan]

I see a problem though, since there is not always just one plot. E.g., in _plot_head_map, there are 3 subplots, which all could have different parameters. Having a general solution could quickly become a mess. If you see any clean way to do it, please post it and we could implement that solution.