hs-zheng / FeatureSelectionGP

Gaussian process regression with feature selection

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

FeatureSelectionGP

Gaussian process regression with elastic net (EN) regularizer for feature selection

This code was used in the following paper :
J. Lim, S. Ha and J. Choi, "Prediction of Reward Functions for Deep Reinforcement Learning via Gaussian Process Regression," in IEEE/ASME Transactions on Mechatronics, vol. 25, no. 4, pp. 1739-1746, Aug. 2020, doi: 10.1109/TMECH.2020.2993564.

Dependency

  • Numpy
  • Scipy
  • Scikit-learn

About

Gaussian process regression with feature selection

License:BSD 3-Clause "New" or "Revised" License


Languages

Language:Python 100.0%