istree.h
上传用户:lengbin
上传日期:2010-03-31
资源大小:121k
文件大小:8k
开发平台:

C/C++

  1. /*----------------------------------------------------------------------
  2.   File    : istree.h
  3.   Contents: item set tree management
  4.   Author  : Christian Borgelt
  5.   History : 22.01.1996 file created
  6.             29.01.1996 ISNODE.offset and ISNODE.id added
  7.             08.02.1996 ISTREE.tacnt, ISTREE.curr, ISTREE.index,
  8.                        ISTREE.head and ISTREE.conf added
  9.             28.03.1996 support made relative to number of item sets
  10.             23.11.1996 ISTREE.levels (first nodes of each level) added
  11.             24.11.1996 ISTREE.arem (add. rule evaluation measure) added
  12.             18.08.1997 chi^2 evaluation measure added
  13.                        parameter 'minlen' added to function ist_init()
  14.             11.02.1998 parameter 'minval' added to function ist_init()
  15.             14.05.1998 item set tree navigation functions added
  16.             08.08.1998 parameter 'apps' added to function ist_create()
  17.             20.08.1998 structure ISNODE redesigned
  18.             07.09.1998 function ist_hedge added
  19.             08.12.1998 function ist_gettac added,
  20.                        float changed to double
  21.             05.02.1999 long int changed to int
  22.             26.08.1999 functions ist_first and ist_last added
  23.             05.11.1999 rule evaluation measure EM_AIMP added
  24.             08.11.1999 parameter 'aval' added to function ist_rule
  25.             01.04.2001 functions ist_set and ist_getcntx added
  26.             28.12.2001 sort function moved to module tract
  27.             07.02.2002 function ist_clear removed, ist_settac added
  28.             11.02.2002 optional use of identifier maps in nodes added
  29.             12.02.2002 ist_first and ist_last replaced by ist_next
  30.             12.03.2003 parameter lift added to function ist_rule
  31.             17.07.2003 functions ist_itemcnt and ist_check added
  32.             18.07.2003 function ist_maxfrq added (item set filter)
  33.             11.08.2003 item set filtering generalized (ist_filter)
  34.             09.05.2004 parameter 'aval' added to function ist_set
  35. ----------------------------------------------------------------------*/
  36. #ifndef __ISTREE__
  37. #define __ISTREE__
  38. #include "tract.h"
  39. /*----------------------------------------------------------------------
  40.   Preprocessor Definitions
  41. ----------------------------------------------------------------------*/
  42. /* --- additional evaluation measures --- */
  43. #define EM_NONE     0           /* no measure */
  44. #define EM_DIFF     1           /* absolute conf. difference to prior */
  45. #define EM_LOGQ     1           /* logarithm of support quotient */
  46. #define EM_QUOT     2           /* difference of conf. quotient to 1 */
  47. #define EM_AIMP     3           /* abs. diff. of improvement to 1 */
  48. #define EM_INFO     4           /* information difference to prior */
  49. #define EM_CHI2     5           /* normalized chi^2 measure */
  50. #define EM_UNKNOWN  6           /* unknown measure */
  51. /* --- item appearances --- */
  52. #define IST_IGNORE  0           /* ignore item */
  53. #define IST_BODY    1           /* item may appear in rule body */
  54. #define IST_HEAD    2           /* item may appear in rule head */
  55. #define IST_BOTH    (IST_HEAD|IST_BODY)
  56. /* --- item set filter modes --- */
  57. #define IST_MAXFRQ  0           /* maximally frequent item sets */
  58. #define IST_CLOSED  1           /* closed item sets */
  59. /*----------------------------------------------------------------------
  60.   Type Definitions
  61. ----------------------------------------------------------------------*/
  62. typedef struct _isnode {        /* --- item set node --- */
  63.   struct _isnode *parent;       /* parent node */
  64.   struct _isnode *succ;         /* successor node on same level */
  65.   int            id;            /* identifier used in parent node */
  66.   int            chcnt;         /* number of child nodes */
  67.   int            size;          /* size   of counter vector */
  68.   int            offset;        /* offset of counter vector */
  69.   int            cnts[1];       /* counter vector */
  70. } ISNODE;                       /* (item set node) */
  71. typedef struct {                /* --- item set tree --- */
  72.   int    tacnt;                 /* number of transactions counted */
  73.   int    lvlvsz;                /* size of level vector */
  74.   int    lvlcnt;                /* number of levels (tree height) */
  75.   ISNODE **levels;              /* first node of each level */
  76.   double supp;                  /* minimal support of an item set */
  77.   double conf;                  /* minimal confidence of a rule */
  78.   int    rsdef;                 /* rule support definition */
  79.   int    arem;                  /* additional rule evaluation measure */
  80.   double minval;                /* minimal evaluation measure value */
  81.   ISNODE *curr;                 /* current node for traversal */
  82.   int    size;                  /* size of item set/rule/hyperedge */
  83.   ISNODE *node;                 /* item set node for extraction */
  84.   int    index;                 /* index in item set node */
  85.   ISNODE *head;                 /* head item node for extraction */
  86.   int    item;                  /* head item of previous rule */
  87.   int    *buf;                  /* buffer for paths (support check) */
  88.   int    *path;                 /* current path / (partial) item set */
  89.   int    plen;                  /* current path length */
  90.   int    hdonly;                /* head only item in current set */
  91.   int    memopt;                /* whether to optimize memory usage */
  92.   int    *map;                  /* to create identifier maps */
  93. #ifdef BENCH                    /* if benchmark version */
  94.   int    cnt;                   /* number of counters */
  95.   int    nec;                   /* number of necessary counters */
  96.   int    chcnt;                 /* number of child pointers */
  97.   int    chnec;                 /* number of necessary child pointers */
  98.   int    bytes;                 /* number of bytes used */
  99. #endif
  100.   char   apps[1];               /* item appearances */
  101. } ISTREE;                       /* (item set tree) */
  102. /*----------------------------------------------------------------------
  103.   Functions
  104. ----------------------------------------------------------------------*/
  105. extern ISTREE* ist_create  (int itemcnt, double supp, double conf,
  106.                             int rsdef, const char *apps, int memopt);
  107. extern void    ist_delete  (ISTREE *ist);
  108. extern int     ist_itemcnt (ISTREE *ist);
  109. extern void    ist_count   (ISTREE *ist, int *set, int cnt);
  110. extern void    ist_countx  (ISTREE *ist, TATREE *tat);
  111. extern int     ist_settac  (ISTREE *ist, int cnt);
  112. extern int     ist_gettac  (ISTREE *ist);
  113. extern int     ist_check   (ISTREE *ist, char *marks);
  114. extern int     ist_addlvl  (ISTREE *ist);
  115. extern int     ist_height  (ISTREE *ist);
  116. extern void    ist_up      (ISTREE *ist, int root);
  117. extern int     ist_down    (ISTREE *ist, int item);
  118. extern int     ist_next    (ISTREE *ist, int item);
  119. extern void    ist_setcnt  (ISTREE *ist, int item, int cnt);
  120. extern int     ist_getcnt  (ISTREE *ist, int item);
  121. extern int     ist_getcntx (ISTREE *ist, int *set, int cnt);
  122. extern void    ist_filter  (ISTREE *ist, int mode);
  123. extern void    ist_init    (ISTREE *ist, int minlen,
  124.                             int arem, double minval);
  125. extern int     ist_set     (ISTREE *ist, int *set,
  126.                             double *supp, double *aval);
  127. extern int     ist_rule    (ISTREE *ist, int *rule,
  128.                             double *supp, double *conf,
  129.                             double *lift, double *aval);
  130. extern int     ist_hedge   (ISTREE *ist, int *hedge,
  131.                             double *supp, double *conf);
  132. #ifndef NDEBUG
  133. extern void    ist_show    (ISTREE *ist);
  134. #endif
  135. /*----------------------------------------------------------------------
  136.   Preprocessor Definitions
  137. ----------------------------------------------------------------------*/
  138. #define ist_itemcnt(t)     ((t)->levels[0]->size)
  139. #define ist_settac(t,n)    ((t)->tacnt = (n))
  140. #define ist_gettac(t)      ((t)->tacnt)
  141. #define ist_height(t)      ((t)->lvlcnt)
  142. #endif