Examples with BoxDataSet edu.cmu.tetrad.data.BoxDataSet used on opensource projects

Example 1 with BoxDataSet

use of edu.cmu.tetrad.data.BoxDataSet in project tetrad by cmu-phil.

the class DataConvertUtils method toMixedDataBox.

public static DataModel toMixedDataBox(MixedTabularDataset mixedTabularDataset) {
    int numOfRows = mixedTabularDataset.getNumOfRows();
    MixedVarInfo[] mixedVarInfos = mixedTabularDataset.getMixedVarInfos();
    double[][] continuousData = mixedTabularDataset.getContinuousData();
    int[][] discreteData = mixedTabularDataset.getDiscreteData();
    List<Node> nodes = new LinkedList<>();
    for (MixedVarInfo mixedVarInfo : mixedVarInfos) {
        if (mixedVarInfo.isContinuous()) {
            nodes.add(new ContinuousVariable(mixedVarInfo.getName()));
        } else {
            nodes.add(new DiscreteVariable(mixedVarInfo.getName(), mixedVarInfo.getCategories()));
        }
    }
    return new BoxDataSet(new MixedDataBox(nodes, numOfRows, continuousData, discreteData), nodes);
}

Example 2 with BoxDataSet

use of edu.cmu.tetrad.data.BoxDataSet in project tetrad by cmu-phil.

the class DataConvertUtils method toContinuousDataModel.

public static DataModel toContinuousDataModel(ContinuousTabularDataset dataset) {
    DataBox dataBox = new DoubleDataBox(dataset.getData());
    List<Node> variables = toNodes(dataset.getVariables());
    return new BoxDataSet(dataBox, variables);
}

Example 3 with BoxDataSet

use of edu.cmu.tetrad.data.BoxDataSet in project tetrad by cmu-phil.

the class DataConvertUtils method toVerticalDiscreteDataModel.

public static DataModel toVerticalDiscreteDataModel(VerticalDiscreteTabularDataset dataset) {
    DataBox dataBox = new VerticalIntDataBox(dataset.getData());
    List<Node> variables = toNodes(dataset.getVariableInfos());
    return new BoxDataSet(dataBox, variables);
}