NextCloud/dlib
1
0
Fork 0
mirror of https://github.com/davisking/dlib.git synced 2024-11-01 10:14:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Made decision functions and segmenter objects callable like normal functions.

Browse Source
This commit is contained in:
Davis King 2013-06-07 23:30:43 -04:00
parent 35a25775d1
commit 97f82b1e4f
3 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
python_examples/sequence_segmenter.py
View File

@ -156,15 +156,15 @@ model = dlib.train_sequence_segmenter(training_sequences, segments, params)
# which are predicted to contain names. If you run this example program you will see that
# it gets them all correct.
for i in range(len(sentences)):
print_segment(sentences[i], model.segment_sequence(training_sequences[i]))
print_segment(sentences[i], model(training_sequences[i]))
# Lets also try segmenting a new sentence. This will print out "Bob Bucket". Note that we
# need to remember to use the same vector representation as we used during training.
test_sentence = "There once was a man from Nantucket whose name rhymed with Bob Bucket"
if use_sparse_vects:
print_segment(test_sentence, model.segment_sequence(sentence_to_sparse_vectors(test_sentence)))
print_segment(test_sentence, model(sentence_to_sparse_vectors(test_sentence)))
else:
print_segment(test_sentence, model.segment_sequence(sentence_to_vectors(test_sentence)))
print_segment(test_sentence, model(sentence_to_vectors(test_sentence)))
# We can also measure the accuracy of a model relative to some labeled data. This
# statement prints the precision, recall, and F1-score of the model relative to the data in

4
tools/python/src/decision_functions.cpp
View File

@ -42,7 +42,7 @@ void add_df (
{
typedef decision_function<kernel_type> df_type;
class_<df_type>(name.c_str())
.def("predict", &predict<df_type>)
.def("__call__", &predict<df_type>)
.def_pickle(serialize_pickle<df_type>());
}
@ -94,7 +94,7 @@ void add_linear_df (
{
typedef decision_function<kernel_type> df_type;
class_<df_type>(name.c_str())
.def("predict", predict<df_type>)
.def("__call__", predict<df_type>)
.add_property("weights", &get_weights<df_type>)
.add_property("bias", get_bias<df_type>, set_bias<df_type>)
.def_pickle(serialize_pickle<df_type>());

4
tools/python/src/sequence_segmenter.cpp
View File

@ -795,8 +795,8 @@ train_sequence_segmenter() and cross_validate_sequence_segmenter() routines. "
.def_pickle(serialize_pickle<segmenter_params>());
class_<segmenter_type> ("segmenter_type")
.def("segment_sequence", &segmenter_type::segment_sequence_dense)
.def("segment_sequence", &segmenter_type::segment_sequence_sparse)
.def("__call__", &segmenter_type::segment_sequence_dense)
.def("__call__", &segmenter_type::segment_sequence_sparse)
.def_readonly("weights", &segmenter_type::get_weights)
.def_pickle(serialize_pickle<segmenter_type>());