7.7.7. mclearn.viz.plot_recall_maps

mclearn.viz.plot_recall_maps(coords_test, y_test, y_pred_test, class_names, output, correct_boolean, vmin=0, vmax=1, mincnt=None, fig_dir='', cmap=<Mock name='mock.pyplot.cm.YlGn' id='140227917342984'>)[source]

Plot the recall map.

Parameters:
  • coords_test (array) – The ra and dec coordinates
  • y_test (array) – The column of predicted values.
  • y_pred_test (array) – The column of predicted values.
  • = array (class_names) – Names of the target (e.g. Galaxy, Star, Quasar)
  • output (str) – The suffix on the saved figure.
  • correct_boolean (array) – A boolean array indicating whehter a test exmaple was correctly predicted.
  • vmin (scalar) – vmin is the value that sits at the bottom end of the colour bar. If None, the min of array C is used.
  • vmax (scalar) – vmax is the value that sits at the top end of the colour bar. If None, the max of array C is used.
  • mincnt (int) – If not None, only display cells with more than mincnt number of points in the cell.
  • cmap (Matplotlib ColorMap object) – The color scheme to be used.