Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
///////////////////////////////////////////////////////////////////////////////
// For information as to what this class does, see the Javadoc, below. //
// Copyright (C) 1998, 1999, 2000, 2001, 2002, 2003, 2004, 2005, 2006, //
// 2007, 2008, 2009, 2010, 2014, 2015, 2022 by Peter Spirtes, Richard //
// Scheines, Joseph Ramsey, and Clark Glymour. //
// //
// This program is free software; you can redistribute it and/or modify //
// it under the terms of the GNU General Public License as published by //
// the Free Software Foundation; either version 2 of the License, or //
// (at your option) any later version. //
// //
// This program is distributed in the hope that it will be useful, //
// but WITHOUT ANY WARRANTY; without even the implied warranty of //
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the //
// GNU General Public License for more details. //
// //
// You should have received a copy of the GNU General Public License //
// along with this program; if not, write to the Free Software //
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA //
///////////////////////////////////////////////////////////////////////////////
package edu.cmu.tetrad.data;
import edu.cmu.tetrad.graph.Node;
import edu.cmu.tetrad.util.NumberFormatUtil;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.Writer;
import java.text.NumberFormat;
import java.util.Iterator;
import java.util.List;
/**
* Provides static methods for saving data to files.
*
* @author josephramsey
*/
public final class DataWriter {
/**
* Writes a dataset to file. The dataset may have continuous and/or discrete columns. Note that out is
* not closed by this method, so the close method on out will need to be called externally.
*
* @param dataSet The data set to save.
* @param out The writer to write the output to.
* @param separator The character separating fields, usually '\t' or ','.
* @throws IOException If there is some problem dealing with the writer.
*/
public static void writeRectangularData(DataSet dataSet,
Writer out, char separator) throws IOException {
NumberFormat nf = NumberFormatUtil.getInstance().getNumberFormat();
StringBuilder buf = new StringBuilder();
for (int col = 0; col < dataSet.getNumColumns(); col++) {
String name = dataSet.getVariable(col).getName();
if (name.trim().equals("")) {
name = "C" + (col - 1);
}
buf.append(name);
if (col < dataSet.getNumColumns() - 1) {
buf.append(separator);
}
}
for (int row = 0; row < dataSet.getNumRows(); row++) {
buf.append("\n");
for (int col = 0; col < dataSet.getNumColumns(); col++) {
Node variable = dataSet.getVariable(col);
if (variable instanceof ContinuousVariable) {
double value = dataSet.getDouble(row, col);
if (ContinuousVariable.isDoubleMissingValue(value)) {
buf.append("*");
} else {
buf.append(nf.format(value));
}
if (col < dataSet.getNumColumns() - 1) {
buf.append(separator);
}
} else if (variable instanceof DiscreteVariable) {
Object obj = dataSet.getObject(row, col);
String val = ((obj == null) ? "" : obj.toString());
buf.append(val);
if (col < dataSet.getNumColumns() - 1) {
buf.append(separator);
}
}
}
}
buf.append("\n");
out.write(buf.toString());
out.close();
}
/**
* Writes the lower triangle of a covariance matrix to file. Note that
* out is not closed by this method, so the close method on
* out will need to be called externally.
*
* @param out The writer to write the output to.
*/
public static void writeCovMatrix(ICovarianceMatrix covMatrix,
PrintWriter out, NumberFormat nf) {
// out.println("/Covariance");
out.println(covMatrix.getSampleSize());
List variables = covMatrix.getVariableNames();
int numVars = variables.size();
int varCount = 0;
for (String variable : variables) {
varCount++;
if (varCount < numVars) {
out.print(variable);
out.print("\t");
} else {
out.println(variable);
}
}
for (int j = 0; j < numVars; j++) {
for (int i = 0; i <= j; i++) {
double value = covMatrix.getValue(i, j);
if (Double.isNaN(value)) {
out.print("*");
} else {
out.print(nf.format(value));
}
out.print((i < j) ? "\t" : "\n");
}
}
out.flush();
out.close();
}
public static void saveKnowledge(Knowledge knowledge, Writer out) throws IOException {
StringBuilder buf = new StringBuilder();
buf.append("/knowledge");
buf.append("\naddtemporal\n");
for (int i = 0; i < knowledge.getNumTiers(); i++) {
String forbiddenWithin = knowledge.isTierForbiddenWithin(i) ? "*" : "";
String onlyCanCauseNextTier = knowledge.isOnlyCanCauseNextTier(i) ? "-" : "";
List tier = knowledge.getTier(i);
if (!tier.isEmpty()) {
buf.append("\n").append(i).append(forbiddenWithin).append(onlyCanCauseNextTier).append(" ");
buf.append(" ");
buf.append(String.join(" ", tier));
}
}
buf.append("\n\nforbiddirect");
for (KnowledgeEdge pair : knowledge.getListOfExplicitlyForbiddenEdges()) {
// for (Iterator i = knowledge.forbiddenEdgesIterator(); i.hasNext(); ) {
// KnowledgeEdge pair = i.next();
String from = pair.getFrom();
String to = pair.getTo();
if (knowledge.isForbiddenByTiers(from, to)) {
continue;
}
buf.append("\n").append(from).append(" ").append(to);
}
buf.append("\n\nrequiredirect");
for (Iterator i
= knowledge.requiredEdgesIterator(); i.hasNext(); ) {
KnowledgeEdge pair = i.next();
String from = pair.getFrom();
String to = pair.getTo();
buf.append("\n").append(from).append(" ").append(to);
}
out.write(buf.toString());
out.flush();
}
}
