Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
dlib
Commits
e35b2d8f
Commit
e35b2d8f
authored
Oct 18, 2017
by
Davis King
Browse files
Added python binary classifier example
parent
525bf0e1
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
66 additions
and
0 deletions
+66
-0
python_examples/svm_binary_classifier.py
python_examples/svm_binary_classifier.py
+66
-0
No files found.
python_examples/svm_binary_classifier.py
0 → 100755
View file @
e35b2d8f
#!/usr/bin/python
# The contents of this file are in the public domain. See LICENSE_FOR_EXAMPLE_PROGRAMS.txt
#
#
# This is an example illustrating the use of a binary SVM classifier tool from
# the dlib C++ Library. In this example, we will create a simple test dataset
# and show how to learn a classifier from it.
#
#
# COMPILING/INSTALLING THE DLIB PYTHON INTERFACE
# You can install dlib using the command:
# pip install dlib
#
# Alternatively, if you want to compile dlib yourself then go into the dlib
# root folder and run:
# python setup.py install
# or
# python setup.py install --yes USE_AVX_INSTRUCTIONS
# if you have a CPU that supports AVX instructions, since this makes some
# things run faster.
#
# Compiling dlib should work on any operating system so long as you have
# CMake and boost-python installed. On Ubuntu, this can be done easily by
# running the command:
# sudo apt-get install libboost-python-dev cmake
#
import
dlib
import
pickle
x
=
dlib
.
vectors
()
y
=
dlib
.
array
()
# Make a training dataset. Here we have just two training examples. Normally
# you would use a much larger training dataset, but for the purpose of example
# this is plenty. For binary classification, the y labels should all be either +1 or -1.
x
.
append
(
dlib
.
vector
([
1
,
2
,
3
,
-
1
,
-
2
,
-
3
]))
y
.
append
(
+
1
)
x
.
append
(
dlib
.
vector
([
-
1
,
-
2
,
-
3
,
1
,
2
,
3
]))
y
.
append
(
-
1
)
# Now make a training object. This object is responsible for turning a
# training dataset into a prediction model. This one here is a SVM trainer
# that uses a linear kernel. If you wanted to use a RBF kernel or histogram
# intersection kernel you could change it to one of these lines:
# svm = dlib.svm_c_trainer_histogram_intersection()
# svm = dlib.svm_c_trainer_radial_basis()
svm
=
dlib
.
svm_c_trainer_linear
()
svm
.
be_verbose
=
True
svm
.
set_c
(
10
)
# Now train the model. The return value is the trained model capable of making predictions.
classifier
=
svm
.
train
(
x
,
y
)
# Now run the model on our data and look at the results.
print
(
"prediction for first sample: {}"
.
format
(
classifier
(
x
[
0
])))
print
(
"prediction for second sample: {}"
.
format
(
classifier
(
x
[
1
])))
# classifier models can also be pickled in the same was as any other python object.
with
open
(
'saved_model.pickle'
,
'wb'
)
as
handle
:
pickle
.
dump
(
classifier
,
handle
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment