Skip to content

Implementation of the Stochastic Gauss-Newton algorithm presented at the International Conference on Machine Learning 2020.

License

Notifications You must be signed in to change notification settings

unc-optimization/SGN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Stochastic Gauss-Newton Methods

Introduction

This package is the implementation of "Stochastic Gauss-Newton Algorithms for Nonconvex Compositional Optimization" including 3 different variants to solve the class of stochastic nonconvex compositional optimization problems. The package includes two numerical examples on compositional optimization problems as presented in the paper. The code is tested using Python 3.7.1.

How to Run

  • First we need the following dependency

pip install pandas scikit-learn matplotlib

  • Then we need to download the datasets. Assuming that we are at SGN_Code folder, now we go to data folder and execute the get_data.sh script cd data sh get_data.sh or download the w8a and covtype datasets at LIBSVM Data and place them in the SGN_Code/data folder.

  • For covtype dataset, we need to decompress the file and rename it from covtype.bz2 to covtype for consistency.

The notebooks to run each example are place in notebook folder. Please refer to them for more details.

Code Usage

We hope that this program will be useful to others, and we would like to hear about your experience with it. If you found it helpful and are using it within our software please cite the following publication:

About

Implementation of the Stochastic Gauss-Newton algorithm presented at the International Conference on Machine Learning 2020.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published