python - scikit を使用した線形回帰予測値のキャッピング

Question

間隔 [0,10] に実数値ラベルを持つデータセットを使用して、線形回帰モデルをトレーニングしています。テスト セットの予測値には、10 を超える予測が含まれています。予測を 10 に制限する方法はありますか。

予測が 10 を超える場合は、明示的に 10 に設定するような条件チェックを行うことを考えています。

より良い方法はありますか？

Answer 1

yが回帰オブジェクトのメソッドの出力である場合、Numpyで 10 にpredict制限できます。minimum

y = np.minimum(y, 10.)

ゼロ以下にもキャップするには、次のようにします。

y = np.maximum(np.minimum(y, 10.), 0.)

または、より短い：

y = np.clip(y, 0., 10.)