-
Notifications
You must be signed in to change notification settings - Fork 2
/
SENNA_VBS.h
46 lines (38 loc) · 985 Bytes
/
SENNA_VBS.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
#ifndef SENNA_VBS_H
#define SENNA_VBS_H
#include "SENNA_Hash.h"
typedef struct SENNA_VBS_
{
/* sizes */
int window_size;
int ll_word_size;
int ll_word_max_idx;
int ll_caps_size;
int ll_caps_max_idx;
int ll_posl_size;
int ll_posl_max_idx;
int input_state_size;
int hidden_state_size;
int output_state_size;
/* weights */
float *ll_word_weight;
float *ll_caps_weight;
float *ll_posl_weight;
float *l1_weight;
float *l1_bias;
float *l2_weight;
float *l2_bias;
/* states */
float *input_state;
float *hidden_state;
float *output_state;
int *labels;
/* padding indices */
int ll_word_padding_idx;
int ll_caps_padding_idx;
int ll_posl_padding_idx;
} SENNA_VBS;
SENNA_VBS* SENNA_VBS_new();
int* SENNA_VBS_forward(SENNA_VBS *vbs, const int *sentence_words, const int *sentence_caps, const int *sentence_posl, int sentence_size);
void SENNA_VBS_free(SENNA_VBS *vbs);
#endif