sul1 permutation

from ai4water import Model

from sul1_utils import get_fitted_model, sul1_data
********** Removing Examples with nan in labels  **********

***** Training *****
input_x shape:  (374, 6)
target shape:  (374, 1)
model = get_fitted_model(Model)
x, y, input_features, output_features = sul1_data()
********** Removing Examples with nan in labels  **********

***** Training *****
input_x shape:  (374, 6)
target shape:  (374, 1)
pimp = model.permutation_importance(x=x,
                                    y=y,
                                    n_repeats=100,
                                    scoring="r2",
                                    plot_type="boxplot")
Base Score 0.153
pimp.plot_1d_pimp("bar_chart")
Base Score 0.153
<AxesSubplot: title={'center': 'Base Score 0.153'}, xlabel='$R^{2}$'>
pimp = model.permutation_importance(x=x,
                                    y=y,
                                    n_repeats=100,
                                    scoring="nse",
                                    plot_type="boxplot")
Base Score 0.083
pimp.plot_1d_pimp("bar_chart")
Base Score 0.083
<AxesSubplot: title={'center': 'Base Score 0.083'}, xlabel='NSE'>
pimp = model.permutation_importance(x=x,
                                    y=y,
                                    n_repeats=100,
                                    scoring="rmsle",
                                    plot_type="boxplot")
Base Score 1.635

Total running time of the script: ( 0 minutes 5.548 seconds)

Gallery generated by Sphinx-Gallery