inference.h

00001 /*
00002  * All of the documentation and software included in the
00003  * Alchemy Software is copyrighted by Stanley Kok, Parag
00004  * Singla, Matthew Richardson, Pedro Domingos, Marc
00005  * Sumner and Hoifung Poon.
00006  * 
00007  * Copyright [2004-07] Stanley Kok, Parag Singla, Matthew
00008  * Richardson, Pedro Domingos, Marc Sumner and Hoifung
00009  * Poon. All rights reserved.
00010  * 
00011  * Contact: Pedro Domingos, University of Washington
00012  * (pedrod@cs.washington.edu).
00013  * 
00014  * Redistribution and use in source and binary forms, with
00015  * or without modification, are permitted provided that
00016  * the following conditions are met:
00017  * 
00018  * 1. Redistributions of source code must retain the above
00019  * copyright notice, this list of conditions and the
00020  * following disclaimer.
00021  * 
00022  * 2. Redistributions in binary form must reproduce the
00023  * above copyright notice, this list of conditions and the
00024  * following disclaimer in the documentation and/or other
00025  * materials provided with the distribution.
00026  * 
00027  * 3. All advertising materials mentioning features or use
00028  * of this software must display the following
00029  * acknowledgment: "This product includes software
00030  * developed by Stanley Kok, Parag Singla, Matthew
00031  * Richardson, Pedro Domingos, Marc Sumner and Hoifung
00032  * Poon in the Department of Computer Science and
00033  * Engineering at the University of Washington".
00034  * 
00035  * 4. Your publications acknowledge the use or
00036  * contribution made by the Software to your research
00037  * using the following citation(s): 
00038  * Stanley Kok, Parag Singla, Matthew Richardson and
00039  * Pedro Domingos (2005). "The Alchemy System for
00040  * Statistical Relational AI", Technical Report,
00041  * Department of Computer Science and Engineering,
00042  * University of Washington, Seattle, WA.
00043  * http://www.cs.washington.edu/ai/alchemy.
00044  * 
00045  * 5. Neither the name of the University of Washington nor
00046  * the names of its contributors may be used to endorse or
00047  * promote products derived from this software without
00048  * specific prior written permission.
00049  * 
00050  * THIS SOFTWARE IS PROVIDED BY THE UNIVERSITY OF WASHINGTON
00051  * AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED
00052  * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
00053  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
00054  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE UNIVERSITY
00055  * OF WASHINGTON OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
00056  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
00057  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
00058  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
00059  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
00060  * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
00061  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
00062  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
00063  * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
00064  * 
00065  */
00066 #ifndef INFERENCE_H_
00067 #define INFERENCE_H_
00068 
00069 #include "variablestate.h"
00070 
00076 class Inference
00077 {
00078  public:
00079 
00091   Inference(VariableState* state, long int seed,
00092             const bool& trackClauseTrueCnts)
00093   {
00094     this->state_ = state;
00095     this->seed_ = seed;
00096     if (seed_ <= 0)
00097     {
00098       struct timeval tv;
00099       struct timezone tzp;
00100       gettimeofday(&tv, &tzp);
00101       seed_ = (long int)((( tv.tv_sec & 0177 ) * 1000000) + tv.tv_usec);
00102     }
00103     srandom(seed_);
00104 
00105     trackClauseTrueCnts_ = trackClauseTrueCnts;
00106     if (trackClauseTrueCnts_)
00107     {
00108         // clauseTrueCnts_ will hold the true counts for each first-order clause
00109       clauseTrueCnts_ = new Array<double>;
00110       clauseTrueCnts_->growToSize(state_->getMLN()->getNumClauses(), 0);
00111     }
00112   }
00113   
00117   virtual ~Inference()
00118   {
00119     if (trackClauseTrueCnts_) delete clauseTrueCnts_;
00120   }
00121 
00125   virtual void init() = 0;
00126 
00130   virtual void infer() = 0;
00131 
00135   virtual void printProbabilities(ostream& out) = 0;
00136   
00140   virtual void printTruePreds(ostream& out) = 0;
00141 
00145   virtual double getProbability(GroundPredicate* const& gndPred) = 0;
00146 
00147   long int getSeed() { return seed_; }
00148   void setSeed(long int s) { seed_ = s; }
00149   
00150   VariableState* getState() { return state_; }
00151   void setState(VariableState* s) { state_ = s; }
00152   
00153   const Array<double>* getClauseTrueCnts() { return clauseTrueCnts_; }
00154 
00155  protected:
00156   
00157     // Seed for randomizer. If not set, then date + time is used
00158   long int seed_;
00159     // State of atoms and clauses used during inference
00160     // Does not belong to inference (do not delete)
00161   VariableState* state_;
00162     // Holds the average number of true groundings of each first-order clause
00163     // in the mln associated with this inference
00164   Array<double>* clauseTrueCnts_;
00165     // Indicates if true counts for each first-order clause are being kept
00166   bool trackClauseTrueCnts_;
00167 };
00168 
00169 #endif /*INFERENCE_H_*/

Generated on Tue Jan 16 05:30:02 2007 for Alchemy by  doxygen 1.5.1