directedmodel.java

来自「这是一个matlab的java实现。里面有许多内容。请大家慢慢捉摸。」· Java 代码 · 共 56 行

/* Copyright (C) 2002 Univ. of Massachusetts Amherst, Computer Science Dept.
   This file is part of "MALLET" (MAchine Learning for LanguagE Toolkit).
   http://www.cs.umass.edu/~mccallum/mallet
   This software is provided under the terms of the Common Public License,
   version 1.0, as published by http://www.opensource.org.  For further
   information, see the file `LICENSE' included with this distribution. */

package edu.umass.cs.mallet.grmm; // Generated package name

import salvo.jesus.graph.DirectedAcyclicGraphImpl;
import java.util.HashMap;
import salvo.jesus.graph.Vertex;


/**
 *  Class for directed graphical models. This is just a
 *   souped-up Graph.
 *
 * Created: Mon Sep 15 14:50:19 2003
 *
 * @author <a href="mailto:[e-mail]">Charles Sutton</a>
 * @version $Id: DirectedModel.java,v 1.1 2004/07/15 17:53:31 casutton Exp $
 */

public class DirectedModel extends DirectedAcyclicGraphImpl implements GraphicalModel {
    
	private HashMap allCpts;

	
	/**
	 *  Returns the conditional distribution <tt>P ( node | Parents (node) )</tt>
	 */
	public DiscretePotential getCpt (Vertex node) {
		return (DiscretePotential) allCpts.get (node);
	}
    
	/**
	 * Sets the conditional distribution of a node.
	 */
	public DiscretePotential setCpt (Vertex node, DiscretePotential dist) {
		return (DiscretePotential) allCpts.put (node, dist);
	}


	public double query (Inferencer inferencer, Assignment assn)
	{
 		return inferencer.query ((DirectedModel)this, assn);
	}

	public void computeMarginals (Inferencer inferencer)
	{
		inferencer.computeMarginals ((DirectedModel)this);
	}

}// DirectedModel