PLearn 0.1
|
00001 // -*- C++ -*- 00002 00003 // PLearn (A C++ Machine Learning Library) 00004 // Copyright (C) 1999-2002 Christian Jauvin 00005 // 00006 00007 // Redistribution and use in source and binary forms, with or without 00008 // modification, are permitted provided that the following conditions are met: 00009 // 00010 // 1. Redistributions of source code must retain the above copyright 00011 // notice, this list of conditions and the following disclaimer. 00012 // 00013 // 2. Redistributions in binary form must reproduce the above copyright 00014 // notice, this list of conditions and the following disclaimer in the 00015 // documentation and/or other materials provided with the distribution. 00016 // 00017 // 3. The name of the authors may not be used to endorse or promote 00018 // products derived from this software without specific prior written 00019 // permission. 00020 // 00021 // THIS SOFTWARE IS PROVIDED BY THE AUTHORS ``AS IS'' AND ANY EXPRESS OR 00022 // IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 00023 // OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN 00024 // NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 00025 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED 00026 // TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 00027 // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 00028 // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 00029 // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 00030 // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 00031 // 00032 // This file is part of the PLearn library. For more information on the PLearn 00033 // library, go to the PLearn Web site at www.plearn.org 00034 00037 #ifndef ProbSparseMatrix_INC 00038 #define ProbSparseMatrix_INC 00039 00040 #include "DoubleAccessSparseMatrix.h" 00041 #include "TMat.h" 00042 namespace PLearn { 00043 using namespace std; 00044 00045 class ProbSparseMatrix : public DoubleAccessSparseMatrix<real> 00046 { 00047 00048 public: 00049 00050 ProbSparseMatrix(int n_rows = 0, int n_cols = 0, string name = "pXY", int mode = ROW_WISE, bool double_access = false); 00051 00052 void incr(int i, int j, real inc = 1.0, bool warning = true); 00053 00054 void set(int i, int j, real value, bool warning = true); 00055 00056 bool checkCondProbIntegrity(); 00057 00058 bool checkJointProbIntegrity(); 00059 00060 void normalizeCond(ProbSparseMatrix& nXY, bool clear_nXY = false); 00061 00062 void normalizeJoint(ProbSparseMatrix& nXY, bool clear_nXY = false); 00063 00064 void normalizeCond(); 00065 00066 void normalizeJoint(); 00067 00068 string getClassName() const { return "ProbSparseMatrix"; } 00069 00070 void iterativeProportionalFittingStep( ProbSparseMatrix& p, Vec& lineMarginal, Vec& colMarginal); 00071 00072 real euclidianDistance( ProbSparseMatrix &p); 00073 00074 void add( ProbSparseMatrix& p, ProbSparseMatrix& q); 00075 00076 }; 00077 00078 template <class T> 00079 inline PStream& operator<<(PStream &out, const ProbSparseMatrix &p) 00080 { 00081 p.write(out); 00082 return out; 00083 } 00084 00085 template <class T> 00086 inline PStream& operator>>(PStream &in, ProbSparseMatrix &p) 00087 { 00088 p.read(in); 00089 return in; 00090 } 00091 00092 class PSMat : public PP<ProbSparseMatrix> 00093 { 00094 00095 public: 00096 00097 PSMat(int n_rows = 0, int n_cols = 0, string name = "pXY", int mode = ROW_WISE, bool double_access = false) : PP<ProbSparseMatrix>(new ProbSparseMatrix(n_rows, n_cols, name, mode, double_access)) {} 00098 00099 PSMat(ProbSparseMatrix* p) : PP<ProbSparseMatrix>(p) {} 00100 00101 }; 00102 00103 } // end of namespace PLearn 00104 00105 #endif 00106 00107 00108 /* 00109 Local Variables: 00110 mode:c++ 00111 c-basic-offset:4 00112 c-file-style:"stroustrup" 00113 c-file-offsets:((innamespace . 0)(inline-open . 0)) 00114 indent-tabs-mode:nil 00115 fill-column:79 00116 End: 00117 */ 00118 // vim: filetype=cpp:expandtab:shiftwidth=4:tabstop=8:softtabstop=4:encoding=utf-8:textwidth=79 :