diff options
| author | ziejd2 | 2017-09-14 16:17:47 -0500 |
|---|---|---|
| committer | GitHub | 2017-09-14 16:17:47 -0500 |
| commit | 7cc31810d53176e805532b2789955f4eedbce6bb (patch) | |
| tree | 82924642070d871f753ee41c0f3e363ff8f380da /BNW_parameter_learning/readInput.m | |
| parent | 6882395afdadf4e982b25b5215071a0932730950 (diff) | |
| download | BNW-7cc31810d53176e805532b2789955f4eedbce6bb.tar.gz | |
Add files via upload
Adding the *.m files used in parameter learning.
Diffstat (limited to 'BNW_parameter_learning/readInput.m')
| -rw-r--r-- | BNW_parameter_learning/readInput.m | 70 |
1 files changed, 70 insertions, 0 deletions
diff --git a/BNW_parameter_learning/readInput.m b/BNW_parameter_learning/readInput.m new file mode 100644 index 00000000..c2c16331 --- /dev/null +++ b/BNW_parameter_learning/readInput.m @@ -0,0 +1,70 @@ +function [ labels, cases, bnet, node_sizes, data,labelsold] = readInput( dfile, sfile, nnodes, std_flag ) + %readInput is to be used when reading in a network with a known structure + % + %Input: + % dfile = name of the file containing the data (required) + % sfile = name of the file containing the structure (required) + % nnodes = number of nodes in the network (required) + % std_flag = flag for whether or not to standardize the data. + % (optional-- Default is FALSE) + % + % See readInputData.m and readInputStructure.m for description of the + % format of the dfile and sfile, respectively. + % + %Output: + % labels = cell array with the names of the nodes. + % cases = cell array with the data. + % bnet = BNT bayesian network with the input structure. + +if nargin < 4 + std_flag = false(1); +end + + +% read in the file with the data +[labelsold,node_sizes,cases, data] = readInputData(dfile,nnodes); + + + + +% read in the file with the structure +[dag] = readInputStructure(sfile,labelsold); + + +% check the ordering of the nodes and reorder if necessary +[labels,cases,dag,node_sizes,ord_flag] = checkStructure(labelsold,cases,dag,node_sizes); +%draw_graph(dag,labels); +%if ord_flag == 1 +% fprintf(['Order of nodes was changed to agree with topological order\n']) +%end +%fprintf(['The structure of the network should be correctly displayed in a figure\n']) + +dcount = 0; +for i = 1:nnodes + if node_sizes(i) ~= 1 + dcount = dcount + 1; + end +end +discrete = zeros(1,dcount); +dcount = 0; +for i = 1:nnodes + if node_sizes(i) ~= 1 + dcount = dcount + 1; + discrete(dcount) = i; + end +end + +bnet = mk_bnet(dag,node_sizes,'discrete',discrete,'names',labels); + +%bnet.dag + +checkDiscreteNodes(bnet,cases); + +% standardize continuous data to have a mean = 0 and std = 1 +if (std_flag) + [cases] = standardizeData(labels,node_sizes,cases); +end + + +end +% end of readInput.m \ No newline at end of file |