Tanl Linguistic Pipeline

classifier/Classifier.h

/*
**  Tanl
**  classifier/Classifier.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_Classifier_H
#define Tanl_Classifier_Classifier_H

#include "include/config.h"
#include "Event.h"

namespace Tanl {
namespace Classifier {

class Classifier
{
public:

  Classifier() :
    verbose(false)
  { }

  Classifier(char const* file);

  Classifier(std::istream& ifs);

  virtual ~Classifier() { }

  ClassID       BestOutcome(double* ocs) const;

  char const*   OutcomeName(int i) const { return outcomeLabels[i]; }

  ClassID       OutcomeID(char const* c) const {
    for (int i = 0; i < outcomeLabels.size(); i++)
      if (!strcmp(outcomeLabels[i], c))
        return i;
    return (ClassID)-1;
  }

  int           NumOutcomes() const { return numOutcomes; }
  
  std::string   PredicateName(int i) const { return predLabels[i]; }

  int           PredicateIndex(char const* name) { return predIndex[name]; }

  void          load(std::istream& is);

  void          save(char const* file);

  Text::WordIndex&      PredIndex() { return predIndex; }

  bool          verbose;

protected:

  std::vector<std::string>      predLabels;
  // associates pred labels to index
  // used by ContextStream when creating contexts
  Text::WordIndex               predIndex;
  // names of event outcomes
  std::vector<char const*>      outcomeLabels;
  // number of outcomes
  ClassID                       numOutcomes;

};

} // namespace Classifier
} // namespace Tanl

#endif // Tanl_Classifier_Classifier_H

 

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