yatsm.plots module

Plots useful for YATSM

yatsm.plots.plot(yatsm, band, freq, ylim=None)

Plot YATSM results for a specified band :param _sphinx_paramlinks_yatsm.plots.plot.yatsm: model :type yatsm: yatsm.YATSM :param _sphinx_paramlinks_yatsm.plots.plot.band: data band to plot :type band: int :param _sphinx_paramlinks_yatsm.plots.plot.freq: frequency of sine/cosine (for predictions) :type freq: iterable :param _sphinx_paramlinks_yatsm.plots.plot.ylim: tuple of floats for y-axes limits :type ylim: tuple

yatsm.plots.plot_crossvalidation_scores(kfold_scores, test_labels)

Plots KFold test summary statistics

Parameters:

• kfold_scores – n by 2 shaped array of mean and standard deviation of KFold scores
• test_labels – n length list of KFold label names

yatsm.plots.plot_feature_importance(algo, cfg)

Plots Random Forest feature importance as barplot

If YATSM configuration (cfg[‘YATSM’]) contains Patsy design information in ‘design_info’, then this plot will show more specific labels for each feature.

Parameters:

• algo – Random Forest algorithm
• cfg – YATSM configuration dictionary