bnlearn-engine.h

Go to the documentation of this file.

#ifndef __BNLEARN_ENGINE_H__
#define __BNLEARN_ENGINE_H__

#include "vfml.h"

typedef enum BNLAction_ {BNL_NO_CHANGE, BNL_ADDED_PARENT, BNL_REMOVED_PARENT, BNL_REVERSED_PARENT} BNLAction;

// Callback API
// Any of these may be NULL, in which case they are ignored
typedef struct BNLearnCallbackAPI_ {  
  // the user data is one-per-Bn
  // action is what action was taken to create the newBn from the oldBn
  void (*NetInit)(void **user, BeliefNet newBn, BeliefNet oldBn, BNLAction action, int childId, int parentId);
  void (*NetFree)(void **user, BeliefNet bn);
  void (*NodeUpdateBD)(void **user, BeliefNetNode bnn,
       float **add_eventCounts, float *add_rowCounts);
} BNLearnCallbackAPI;


/* The various parameters you may specify to bnlearn, and in
   comments are the default values */
typedef struct BNLearnParams_ {
   // Data Input. One of these should be non-NULL
   VoidListPtr gDataMemory;   // Fastest. Consider using ExamplesRead(...)
   FILE *gDataFile;           // Data is loaded from disk on each iteration. Useful if data does not fit into RAM

   // Input Network. Exactly one of these needs to be non-NULL.
   BeliefNet gInputNetMemory;
   FILE *gInputNetFile;
   ExampleSpecPtr gInputNetEmptySpec; // consider using ExampleSpecRead(char *fileName);

   // Results
   char *gOutputNetFilename;     // name of file to output to. NULL for none
   int gOutputNetToMemory;       // Boolean (defaults to false)
   BeliefNet gOutputNetMemory;   // holds the resulting network upon return

   // Parameters
   long  gLimitBytes;              /*   -1      */
   double gLimitSeconds;           /*   -1      */
   int   gNoReverse;               /*   0       */
   double gKappa;                  /*   0.5     */
   int   gOnlyEstimateParameters;  /*   0       */
   int   gMaxSearchSteps;          /*   -1      */
   int   gMaxParentsPerNode;       /*   -1      */
   int   gMaxParameterGrowthMult;  /*   -1      */
   long  gMaxParameterCount;       /*   -1      */
   int   gSeed;                    /*   -1      */
   double gSmoothAmount;           /*   1.0     */
   int gCheckModelCycle;           /*  0        */
   BNLearnCallbackAPI *gCallbackAPI; /* NULL    */
} BNLearnParams;



BNLearnParams *BNLearn_NewParams();

void BNLearn_FreeParams(BNLearnParams *params);

void BNLearn(BNLearnParams *params);  


#endif

---

Generated for VFML by hosted by