use of edu.pitt.dbmi.data.reader.tabular.MixedVarInfo 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);
}
Aggregations