Tanl Linguistic Pipeline

classifier/LBFGS.h

/*
**  Tanl
**  classifier/LBFGS.h
**  ----------------------------------------------------------------------
**  Copyright (c) 2003  Giuseppe Attardi (attardi@di.unipi.it).
**  ----------------------------------------------------------------------
**
**  This file is part of Tanl.
**
**  Tanl is free software; you can redistribute it and/or modify it
**  under the terms of the GNU General Public License, version 3,
**  as published by the Free Software Foundation.
**
**  Tanl is distributed in the hope that it will be useful,
**  but WITHOUT ANY WARRANTY; without even the implied warranty of
**  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
**  GNU General Public License for more details.
**
**  You should have received a copy of the GNU General Public License
**  along with this program.  If not, see <http://www.gnu.org/licenses/>.
**  ----------------------------------------------------------------------
*/

#ifndef Tanl_Classifier_LBFGS_H
#define Tanl_Classifier_LBFGS_H

#include <float.h>              // DBL_EPSILON

#include "MaxEnt.h"

namespace Tanl {
namespace Classifier {

typedef std::vector<std::vector<std::pair<ClassID, int> > > ContribTable;

struct  Options {
  int     n;                    
  int     m;                    
  int     niter;                
  int     nfuns;                
  int     iflag;
  int     iprint[2]; 
  double  eps;          
  double  xtol;
  int       diagco;
  double* diag;
  double* w;

  Options(int n, double eps) :
    n(0),
    m(5),                       // 3 <= m <= 7, see LBFGS_FUN()
    niter(0),
    nfuns(0),
    iflag(0),
    eps(eps),
    xtol(DBL_EPSILON),
    diagco(0),
    diag(0),
    w(0)
  {
    iprint[0] = -1;       // by default print nothing
    iprint[1] = 0;
  }

  ~Options() {
    delete[] w;
    delete[] diag;
  }
};

class LBFGS : public MaxEnt
{
public:
  LBFGS(EventStream& es, int iterations, int cutoff = 0, double eps = 1E-05);

  LBFGS(int iterations = 50, int cutoff = 0, double eps = 1E-05);

  ~LBFGS() {
    delete[] lambda;
  }

  void  train();

  void  train(EventStream& es);

  void  save(char const* path);

  void  writeHeader(std::ofstream& ofs);
  void  writeData(std::ofstream& ofs);

protected:

  double*       lambda; // overrides definition in MaxEnt

private:
  ClassID       estimate(const std::vector<PID>& preds, double probs[]);

  int                   numEvents; 

  Options               opt;    
  ContribTable  contribTable; 

};

} // namespace Classifier
} // namespace Tanl

#endif // Tanl_Classifier_LBFGS_H

 

Copyright © 2005-2011 G. Attardi. Generated on 4 Mar 2011 by doxygen 1.6.1.