Adjusting Plots¶
NannyML uses Plotly [1] to generate figures and returns plotly.graph_objs._figure.Figure
from .plot
methods.
When you need the plot to do something other than the default plotting, e.g. add another curve or indicate a specific
time period in the figure, you can do this by updating the plot.
The example below adds an additional indicator for a particular period of interest using this method.
>>> import nannyml as nml
>>> reference, analysis, analysis_target = nml.load_synthetic_binary_classification_dataset()
>>> estimator = nml.CBPE(
... y_pred_proba='y_pred_proba',
... y_pred='y_pred',
... y_true='work_home_actual',
... timestamp_column_name='timestamp',
... metrics=['roc_auc'],
... chunk_size=5000,
... problem_type='classification_binary',
>>> ).fit(reference)
>>> estimated_performance = estimator.estimate(analysis)
>>> figure = estimated_performance.plot(kind='performance')
>>> # indicate period of interest
>>> import datetime as dt
>>> figure.add_vrect(x0=dt.datetime(2019,1,1), x1=dt.datetime(2020, 1,1),
... annotation_text="Strategy change", annotation_position="top left")
>>> figure.show()