File tree Expand file tree Collapse file tree 5 files changed +863
-1
lines changed Expand file tree Collapse file tree 5 files changed +863
-1
lines changed Original file line number Diff line number Diff line change @@ -6,6 +6,7 @@ py_library(
6
6
name = "text" ,
7
7
srcs = ([
8
8
"__init__.py" ,
9
+ "crf.py" ,
9
10
"skip_gram_ops.py" ,
10
11
]),
11
12
data = [
@@ -15,6 +16,19 @@ py_library(
15
16
srcs_version = "PY2AND3" ,
16
17
)
17
18
19
+ py_test (
20
+ name = "crf_test" ,
21
+ size = "small" ,
22
+ srcs = [
23
+ "crf_test.py" ,
24
+ ],
25
+ main = "crf_test.py" ,
26
+ srcs_version = "PY2AND3" ,
27
+ deps = [
28
+ ":text" ,
29
+ ],
30
+ )
31
+
18
32
py_test (
19
33
name = "skip_gram_ops_test" ,
20
34
size = "small" ,
Original file line number Diff line number Diff line change 4
4
| Submodule | Maintainers | Contact Info |
5
5
| :---------- | :----------- | :------------- |
6
6
| skip_gram_ops | | |
7
+ | crf
| Dheeraj R. Reddy
| [email protected] |
7
8
8
9
## Components
9
10
| Submodule | Text Processing Function | Reference |
Original file line number Diff line number Diff line change 17
17
from __future__ import division
18
18
from __future__ import print_function
19
19
20
+ # Conditional Random Field
21
+ from tensorflow_addons .text .crf import crf_binary_score
22
+ from tensorflow_addons .text .crf import crf_decode
23
+ from tensorflow_addons .text .crf import crf_decode_backward
24
+ from tensorflow_addons .text .crf import crf_decode_forward
25
+ from tensorflow_addons .text .crf import crf_forward
26
+ from tensorflow_addons .text .crf import crf_log_likelihood
27
+ from tensorflow_addons .text .crf import crf_log_norm
28
+ from tensorflow_addons .text .crf import crf_multitag_sequence_score
29
+ from tensorflow_addons .text .crf import crf_sequence_score
30
+ from tensorflow_addons .text .crf import crf_unary_score
31
+ from tensorflow_addons .text .crf import viterbi_decode
32
+
20
33
# Skip Gram Sampling
21
34
from tensorflow_addons .text .skip_gram_ops import skip_gram_sample
22
- from tensorflow_addons .text .skip_gram_ops import skip_gram_sample_with_text_vocab
35
+ from tensorflow_addons .text .skip_gram_ops import skip_gram_sample_with_text_vocab
You can’t perform that action at this time.
0 commit comments