def __init__(self, network, datastore: bayesianpy.data.DataSet, logger: logging.Logger): self._logger = logger self._datastore = datastore # serialise the network as a string. if isinstance(network, bayesianpy.network.Network): self._network = network.to_xml() else: from xml.dom import minidom nt = network.saveToString() reparsed = minidom.parseString(nt) self._network = reparsed.toprettyxml(indent=" ")
def __init__(self, network, datastore: bayesianpy.data.DaskDataset): self._logger = logging.getLogger(__name__) self._datastore = datastore # serialise the network as a string. if isinstance(network, bayesianpy.network.Network): self._network = network.to_xml() else: from xml.dom import minidom nt = network.saveToString() reparsed = minidom.parseString(nt) self._network = reparsed.toprettyxml(indent=" ") if not isinstance(datastore.get_dataframe(), dd.DataFrame): raise ValueError("Dataframe has to be of type Dask.DataFrame")