evalb.h

Go to the documentation of this file.

#ifndef META_PARSER_EVALB_H_
#define META_PARSER_EVALB_H_

#include "parser/trees/parse_tree.h"

namespace meta
{
namespace parser
{

class evalb
{
  public:
    uint64_t matched() const
    {
        return proposed_correct_;
    }

    uint64_t proposed_total() const
    {
        return proposed_total_;
    }

    uint64_t gold_total() const
    {
        return gold_total_;
    }

    double labeled_precision() const
    {
        return static_cast<double>(matched()) / proposed_total() * 100;
    }

    double labeled_recall() const
    {
        return static_cast<double>(matched()) / gold_total() * 100;
    }

    double labeled_f1() const
    {
        return 2 * (labeled_precision() * labeled_recall())
               / (labeled_precision() + labeled_recall());
    }

    double perfect() const
    {
        return static_cast<double>(perfect_) / total_trees_ * 100;
    }

    double average_crossing() const
    {
        return static_cast<double>(crossed_) / total_trees_;
    }

    double zero_crossing() const
    {
        return static_cast<double>(zero_crossing_) / total_trees_ * 100;
    }

    void add_tree(parse_tree proposed, parse_tree gold);

  private:
    uint64_t proposed_correct_ = 0;

    uint64_t proposed_total_ = 0;

    uint64_t gold_total_ = 0;

    uint64_t perfect_ = 0;

    uint64_t crossed_ = 0;

    uint64_t zero_crossing_ = 0;

    uint64_t total_trees_ = 0;
};
}
}
#endif