BaggingMachine.h

Go to the documentation of this file.

/*
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * Written (W) 2013 Viktor Gal
 * Copyright (C) 2013 Viktor Gal
 */

#ifndef BAGGINGMACHINE_H
#define BAGGINGMACHINE_H

#include <shogun/machine/Machine.h>
#include <shogun/ensemble/CombinationRule.h>
#include <shogun/evaluation/Evaluation.h>

namespace shogun
{
    class CBaggingMachine : public CMachine
    {
        public:
            CBaggingMachine();

            CBaggingMachine(CFeatures* features, CLabels* labels);

            virtual ~CBaggingMachine();

            virtual CBinaryLabels* apply_binary(CFeatures* data=NULL);
            virtual CMulticlassLabels* apply_multiclass(CFeatures* data=NULL);
            virtual CRegressionLabels* apply_regression(CFeatures* data=NULL);

            void set_num_bags(int32_t num_bags);

            int32_t get_num_bags() const;

            void set_bag_size(int32_t bag_size);

            int32_t get_bag_size() const;

            CMachine* get_machine() const;

            void set_machine(CMachine* machine);

            void set_combination_rule(CCombinationRule* rule);
            
            CCombinationRule* get_combination_rule() const;

            virtual EMachineType get_classifier_type() { return CT_BAGGING; }

            float64_t get_oob_error(CEvaluation* eval) const;

            virtual const char* get_name() const { return "BaggingMachine"; }

        protected:
            virtual bool train_machine(CFeatures* data=NULL);

            SGVector<float64_t> apply_get_outputs(CFeatures* data);

        private:
            void register_parameters();
            void init();

            CDynamicArray<index_t>* get_oob_indices(const SGVector<index_t>& in_bag);

            void clear_oob_indicies();

        private:
            CDynamicObjectArray* m_bags;

            CFeatures* m_features;

            CMachine* m_machine;

            int32_t m_num_bags;

            int32_t m_bag_size;

            CCombinationRule* m_combination_rule;

            SGVector<bool> m_all_oob_idx;

            CDynamicObjectArray* m_oob_indices;
    };
}

#endif /* BAGGINGMACHINE_H */