PLearn 0.1
|
00001 // -*- C++ -*- 00002 00003 // SecondIterationWrapper.h 00004 // Copyright (c) 1998-2002 Pascal Vincent 00005 // Copyright (C) 1999-2002 Yoshua Bengio and University of Montreal 00006 // Copyright (c) 2002 Jean-Sebastien Senecal, Xavier Saint-Mleux, Rejean Ducharme 00007 // 00008 // Redistribution and use in source and binary forms, with or without 00009 // modification, are permitted provided that the following conditions are met: 00010 // 00011 // 1. Redistributions of source code must retain the above copyright 00012 // notice, this list of conditions and the following disclaimer. 00013 // 00014 // 2. Redistributions in binary form must reproduce the above copyright 00015 // notice, this list of conditions and the following disclaimer in the 00016 // documentation and/or other materials provided with the distribution. 00017 // 00018 // 3. The name of the authors may not be used to endorse or promote 00019 // products derived from this software without specific prior written 00020 // permission. 00021 // 00022 // THIS SOFTWARE IS PROVIDED BY THE AUTHORS ``AS IS'' AND ANY EXPRESS OR 00023 // IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 00024 // OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN 00025 // NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 00026 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED 00027 // TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 00028 // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 00029 // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 00030 // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 00031 // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 00032 // 00033 // This file is part of the PLearn library. For more information on the PLearn 00034 // library, go to the PLearn Web site at www.plearn.org 00035 00036 00037 /* ******************************************************************************** 00038 * $Id: SecondIterationWrapper.h, v 1.0 2004/07/19 10:00:00 Bengio/Kegl/Godbout * 00039 * This file is part of the PLearn library. * 00040 ******************************************************************************** */ 00041 00044 #ifndef SecondIterationWrapper_INC 00045 #define SecondIterationWrapper_INC 00046 00047 #include <plearn_learners/generic/PLearner.h> 00048 #include <plearn/base/stringutils.h> 00049 00050 namespace PLearn { 00051 using namespace std; 00052 00053 class SecondIterationWrapper: public PLearner 00054 { 00055 typedef PLearner inherited; 00056 00057 private: 00058 00059 /* 00060 Build options: they have to be set before training 00061 */ 00062 00063 int class_prediction; 00064 VMat ref_train; 00065 VMat ref_test; 00066 VMat ref_sales; 00067 VMat train_dataset; 00068 VMat test_dataset; 00069 PP<PLearner> base_regressor_template; // template for a generic regressor as the base learner to be boosted 00070 00071 /* 00072 Learnt options: they are sized and initialized if need be, at stage 0 00073 */ 00074 00075 00076 PP<PLearner> base_regressor; // base regressors built at each boosting stage 00077 00078 /* 00079 Work fields: they are sized and initialized if need be, at buid time 00080 */ 00081 int margin; 00082 int loan; 00083 int sales; 00084 int tclass; 00085 Mat search_table; 00086 00087 Vec sample_input; 00088 Vec sample_target; 00089 Vec sample_output; 00090 Vec sample_costs; 00091 00092 00093 public: 00094 SecondIterationWrapper(); 00095 virtual ~SecondIterationWrapper(); 00096 00097 PLEARN_DECLARE_OBJECT(SecondIterationWrapper); 00098 00099 static void declareOptions(OptionList& ol); 00100 virtual void makeDeepCopyFromShallowCopy(CopiesMap &copies); 00101 virtual void build(); 00102 virtual void train(); 00103 virtual void forget(); 00104 virtual int outputsize() const; 00105 virtual TVec<string> getTrainCostNames() const; 00106 virtual TVec<string> getTestCostNames() const; 00107 virtual void computeOutput(const Vec& input, Vec& output) const; 00108 virtual void computeOutputAndCosts(const Vec& input, const Vec& target, Vec& output, Vec& costs) const; 00109 virtual void computeCostsFromOutputs(const Vec& input, const Vec& output, const Vec& target, Vec& costs) const; 00110 00111 private: 00112 void build_(); 00113 void computeClassStatistics(); 00114 void computeSalesStatistics(); 00115 real deGaussianize(real prediction); 00116 }; 00117 00118 DECLARE_OBJECT_PTR(SecondIterationWrapper); 00119 00120 } // end of namespace PLearn 00121 00122 #endif 00123 00124 00125 /* 00126 Local Variables: 00127 mode:c++ 00128 c-basic-offset:4 00129 c-file-style:"stroustrup" 00130 c-file-offsets:((innamespace . 0)(inline-open . 0)) 00131 indent-tabs-mode:nil 00132 fill-column:79 00133 End: 00134 */ 00135 // vim: filetype=cpp:expandtab:shiftwidth=4:tabstop=8:softtabstop=4:encoding=utf-8:textwidth=79 :