ramjet.analysis.lightcurve_visualizer

Code for visualizing lightcurves.

Module Contents

plot_lightcurve(times: np.ndarray, fluxes: np.ndarray, labels: np.ndarray = None, predictions: np.ndarray = None, title: str = None, x_label: str = 'Days', y_label: str = 'Flux', x_limits: (float, float) = (None, None), y_limits: (float, float) = (None, None), save_path: Union[Path, str] = None)

Plots a lightcurve with a consistent styling. If true labels and/or predictions are included, these will additionally be plotted.

Parameters:

• times – The times of the measurements.
• fluxes – The fluxes of the measurements.
• labels – The binary labels for each time step.
• predictions – The probability prediction for each time step.
• title – The title of the plot.
• x_label – The label for the x axis.
• y_label – The label for the y axis.
• x_limits – Optional axis limiting for the x axis.
• y_limits – Optional axis limiting for the y axis.
• save_path – The path to save the plot to. If None, the plot will be shown instead.