Package edu.cmu.tetrad.search.work_in_progress

Class MixtureModel

java.lang.Object
edu.cmu.tetrad.search.work_in_progress.MixtureModel
public class MixtureModel extends Object
Represents a Gaussian mixture model -- a dataset with data sampled from two or more multivariate Gaussian distributions.
Version:
$Id: $Id
Author:
Madelyn Glymour

  • Constructor Summary

    Constructors
    Constructor
    Description
    MixtureModel(DataSet data, double[][] dataArray, double[][] meansArray, double[] weightsArray, Matrix[] variancesArray, double[][] gammaArray)
    Constructs a mixture model from a mixed data set, a means matrix, a weights array, a variance matrix, and a gamma matrix.

  • Method Summary

    Modifier and Type
    Method
    Description
    int[]
    getCases()
    Getter for the field cases.
    double[][]
    getData()
    Getter for the field data.
    DataSet[]
    getDemixedData()
    getDemixedData.
    int
    getDistribution(int caseNum)
    Classifies a given case into a model, based on which model has the highest gamma value for that case.
    double[][]
    getMeans()
    getMeans.
    Matrix[]
    getVariances()
    getVariances.
    double[]
    getWeights()
    getWeights.
    double[]
    searchDemixedData()
    Perform an FGES search on each of the demixed data sets.

    Methods inherited from class java.lang.Object

    equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

  • Constructor Details

    • MixtureModel

      public MixtureModel(DataSet data, double[][] dataArray, double[][] meansArray, double[] weightsArray, Matrix[] variancesArray, double[][] gammaArray)
      Constructs a mixture model from a mixed data set, a means matrix, a weights array, a variance matrix, and a gamma matrix.
      Parameters:
      data - the mixed data set
      dataArray - the mixed data set in array form
      meansArray - the means matrix
      weightsArray - the weights array
      variancesArray - the variance matrix
      gammaArray - the gamma matrix

  • Method Details

    • getData

      public double[][] getData()

      Getter for the field data.

      Returns:
      the mixed data set in array form

    • getMeans

      public double[][] getMeans()

      getMeans.

      Returns:
      the means matrix

    • getWeights

      public double[] getWeights()

      getWeights.

      Returns:
      the weights array

    • getVariances

      public Matrix[] getVariances()

      getVariances.

      Returns:
      the variance matrix

    • getCases

      public int[] getCases()

      Getter for the field cases.

      Returns:
      an array assigning each case an integer corresponding to a model

    • getDistribution

      public int getDistribution(int caseNum)
      Classifies a given case into a model, based on which model has the highest gamma value for that case.
      Parameters:
      caseNum - a int
      Returns:
      a int

    • getDemixedData

      public DataSet[] getDemixedData()

      getDemixedData.

      Returns:
      an array of DataSet objects

    • searchDemixedData

      public double[] searchDemixedData() throws InterruptedException
      Perform an FGES search on each of the demixed data sets.
      Returns:
      the BIC scores of the graphs returned by searches.
      Throws:
      InterruptedException - if any.