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(
66 name = "text" ,
77 srcs = ([
88 "__init__.py" ,
9+ "crf.py" ,
910 "skip_gram_ops.py" ,
1011 ]),
1112 data = [
@@ -15,6 +16,19 @@ py_library(
1516 srcs_version = "PY2AND3" ,
1617)
1718
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+
1832py_test (
1933 name = "skip_gram_ops_test" ,
2034 size = "small" ,
Original file line number Diff line number Diff line change 44| Submodule | Maintainers | Contact Info |
55| :---------- | :----------- | :------------- |
66| skip_gram_ops | | |
7+ | crf
| Dheeraj R. Reddy
| [email protected] | 78
89## Components
910| Submodule | Text Processing Function | Reference |
Original file line number Diff line number Diff line change 1717from __future__ import division
1818from __future__ import print_function
1919
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+
2033# Skip Gram Sampling
2134from 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