yatsm.plots module¶
Plots useful for YATSM
-
yatsm.plots.
plot
(yatsm, band, freq, ylim=None)[source]¶ Plot YATSM results for a specified band :param yatsm: model :type yatsm: yatsm.YATSM :param band: data band to plot :type band: int :param freq: frequency of sine/cosine (for predictions) :type freq: iterable :param ylim: tuple of floats for y-axes limits :type ylim: tuple
-
yatsm.plots.
plot_crossvalidation_scores
(kfold_scores, test_labels)[source]¶ Plots KFold test summary statistics
Parameters: - kfold_scores (np.ndarray) – n by 2 shaped array of mean and standard deviation of KFold scores
- test_labels (list) – n length list of KFold label names
-
yatsm.plots.
plot_feature_importance
(algo, cfg)[source]¶ 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 (sklearn.ensemble.RandomForestClassifier) – Random Forest algorithm
- cfg (dict) – YATSM configuration dictionary