drwnRegression.h

/******************************************************************************
** DARWIN: A FRAMEWORK FOR MACHINE LEARNING RESEARCH AND DEVELOPMENT
** Distributed under the terms of the BSD license (see the LICENSE file)
** Copyright (c) 2007-2015, Stephen Gould
** All rights reserved.
**
******************************************************************************
** FILENAME:    drwnRegression.h
** AUTHOR(S):   Stephen Gould <stephen.gould@anu.edu.au>
**
*****************************************************************************/

#pragma once

#include <cstdlib>
#include <vector>

#include "drwnBase.h"
#include "drwnDataset.h"

using namespace std;

// drwnRegression -----------------------------------------------------------

class drwnRegression : public drwnStdObjIface, public drwnProperties {
 protected:
    int _nFeatures;     
    bool _bValid;       

 public:
    drwnRegression();
    drwnRegression(unsigned n);
    drwnRegression(const drwnRegression &r);
    virtual ~drwnRegression() {
        // do nothing
    }

    // access functions
    int numFeatures() const { return _nFeatures; }
    virtual bool valid() const { return _bValid; }

    // initialization
    virtual void initialize(unsigned n);

    // i/o
    virtual bool save(drwnXMLNode& xml) const;
    virtual bool load(drwnXMLNode& xml);

    // training
    virtual double train(const drwnRegressionDataset& dataset) = 0;
    virtual double train(const vector<vector<double> >& features,
        const vector<double>& targets);
    virtual double train(const vector<vector<double> >& features,
        const vector<double>& targets, const vector<double>& weights);
    virtual double train(const char *filename);

    // evaluation (regression)
    virtual double getRegression(const vector<double>& features) const = 0;
    virtual void getRegressions(const vector<vector<double> >& features,
        vector<double>& outputTargets) const;
};