forked from Craigacp/MIToolbox
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathCalculateProbability.h
100 lines (87 loc) · 3.47 KB
/
CalculateProbability.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
/*******************************************************************************
** CalculateProbability.h
** Part of the mutual information toolbox
**
** Contains functions to calculate the probability of each state in the array
** and to calculate the probability of the joint state of two arrays
**
** Author: Adam Pocock
** Created 17/2/2010
**
** Copyright 2010 Adam Pocock, The University Of Manchester
** www.cs.manchester.ac.uk
**
** This file is part of MIToolbox, licensed under the 3-clause BSD license.
*******************************************************************************/
#ifndef __CalculateProbability_H
#define __CalculateProbability_H
#ifdef __cplusplus
extern "C" {
#endif
typedef struct jpState
{
double *jointProbabilityVector;
int numJointStates;
double *firstProbabilityVector;
int numFirstStates;
double *secondProbabilityVector;
int numSecondStates;
} JointProbabilityState;
typedef struct pState
{
double *probabilityVector;
int numStates;
} ProbabilityState;
typedef struct wjpState
{
double *jointProbabilityVector;
double *jointWeightVector;
int numJointStates;
double *firstProbabilityVector;
double *firstWeightVector;
int numFirstStates;
double *secondProbabilityVector;
double *secondWeightVector;
int numSecondStates;
} WeightedJointProbState;
typedef struct wpState
{
double *probabilityVector;
double *stateWeightVector;
int numStates;
} WeightedProbState;
/*******************************************************************************
** calculateJointProbability returns the joint probability vector of two vectors
** and the marginal probability vectors in a struct.
** It is the base operation for all information theory calculations involving
** two or more variables.
**
** length(firstVector) == length(secondVector) == vectorLength otherwise there
** will be a segmentation fault
*******************************************************************************/
JointProbabilityState calculateJointProbability(double *firstVector, double *secondVector, int vectorLength);
/*******************************************************************************
** calculateWeightedJointProbability returns the same joint state object
** as calculateJointProbability, with additional weightVectors giving the
** weight assigned to each state.
*******************************************************************************/
WeightedJointProbState calculateWeightedJointProbability(double *firstVector, double *secondVector, double *exampleWeightVector, int vectorLength);
/*******************************************************************************
** calculateProbability returns the probability vector from one vector.
** It is the base operation for all information theory calculations involving
** one variable
**
** length(dataVector) == vectorLength otherwise there
** will be a segmentation fault
*******************************************************************************/
ProbabilityState calculateProbability(double *dataVector, int vectorLength);
/*******************************************************************************
** calculateWeightedProbability returns the same state object
** as calculateProbability, with an additional weightVector giving the
** weight assigned to each state.
*******************************************************************************/
WeightedProbState calculateWeightedProbability(double *dataVector, double *exampleWeightVector, int vectorLength);
#ifdef __cplusplus
}
#endif
#endif