/** {@inheritDoc} */
@Override
public final void save(final OutputStream os, final Object obj) {
final EncogWriteHelper out = new EncogWriteHelper(os);
final BayesianNetwork b = (BayesianNetwork) obj;
out.addSection("BAYES-NETWORK");
out.addSubSection("BAYES-PARAM");
String queryType = "";
String queryStr = b.getClassificationStructure();
if (b.getQuery() != null) {
queryType = b.getQuery().getClass().getSimpleName();
}
out.writeProperty("queryType", queryType);
out.writeProperty("query", queryStr);
out.writeProperty("contents", b.getContents());
out.addSubSection("BAYES-PROPERTIES");
out.addProperties(b.getProperties());
out.addSubSection("BAYES-TABLE");
for (BayesianEvent event : b.getEvents()) {
for (TableLine line : event.getTable().getLines()) {
if (line == null) continue;
StringBuilder str = new StringBuilder();
str.append("P(");
str.append(BayesianEvent.formatEventName(event, line.getResult()));
if (event.getParents().size() > 0) {
str.append("|");
}
int index = 0;
boolean first = true;
for (BayesianEvent parentEvent : event.getParents()) {
if (!first) {
str.append(",");
}
first = false;
int arg = line.getArguments()[index++];
if (parentEvent.isBoolean()) {
if (arg == 0) {
str.append("+");
} else {
str.append("-");
}
}
str.append(parentEvent.getLabel());
if (!parentEvent.isBoolean()) {
str.append("=");
if (arg >= parentEvent.getChoices().size()) {
throw new BayesianError(
"Argument value " + arg + " is out of range for event " + parentEvent.toString());
}
str.append(parentEvent.getChoice(arg));
}
}
str.append(")=");
str.append(line.getProbability());
str.append("\n");
out.write(str.toString());
}
}
out.flush();
}
