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
66c4fa81
"examples/pytorch/vscode:/vscode.git/clone" did not exist on "02e5d47d2a247e1a12f22c137683fce0b187d3bf"
Commit
66c4fa81
authored
Jun 03, 2018
by
Davis King
Browse files
Made decision functions more viewable from python.
parent
106c905b
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
33 additions
and
0 deletions
+33
-0
tools/python/src/decision_functions.cpp
tools/python/src/decision_functions.cpp
+33
-0
No files found.
tools/python/src/decision_functions.cpp
View file @
66c4fa81
...
...
@@ -45,6 +45,15 @@ void add_df (
typedef
decision_function
<
kernel_type
>
df_type
;
py
::
class_
<
df_type
>
(
m
,
name
.
c_str
())
.
def
(
"__call__"
,
&
predict
<
df_type
>
)
.
def_property_readonly
(
"alpha"
,
[](
const
df_type
&
df
)
{
return
df
.
alpha
;})
.
def_property_readonly
(
"b"
,
[](
const
df_type
&
df
)
{
return
df
.
b
;})
.
def_property_readonly
(
"kernel_function"
,
[](
const
df_type
&
df
)
{
return
df
.
kernel_function
;})
.
def_property_readonly
(
"basis_vectors"
,
[](
const
df_type
&
df
)
{
std
::
vector
<
matrix
<
double
,
0
,
1
>>
temp
;
for
(
long
i
=
0
;
i
<
df
.
basis_vectors
.
size
();
++
i
)
temp
.
push_back
(
sparse_to_dense
(
df
.
basis_vectors
(
i
)));
return
temp
;
})
.
def
(
py
::
pickle
(
&
getstate
<
df_type
>
,
&
setstate
<
df_type
>
));
}
...
...
@@ -105,6 +114,22 @@ void add_linear_df (
// ----------------------------------------------------------------------------------------
std
::
string
radial_basis_kernel__repr__
(
const
radial_basis_kernel
<
sample_type
>&
item
)
{
std
::
ostringstream
sout
;
sout
<<
"radial_basis_kernel(gamma="
<<
item
.
gamma
<<
")"
;
return
sout
.
str
();
}
std
::
string
linear_kernel__repr__
(
const
linear_kernel
<
sample_type
>&
item
)
{
std
::
ostringstream
sout
;
sout
<<
"linear_kernel()"
;
return
sout
.
str
();
}
// ----------------------------------------------------------------------------------------
std
::
string
binary_test__str__
(
const
binary_test
&
item
)
{
std
::
ostringstream
sout
;
...
...
@@ -170,6 +195,14 @@ void bind_decision_functions(py::module &m)
add_df
<
polynomial_kernel
<
sample_type
>
>
(
m
,
"_decision_function_polynomial"
);
add_df
<
sparse_polynomial_kernel
<
sparse_vect
>
>
(
m
,
"_decision_function_sparse_polynomial"
);
py
::
class_
<
radial_basis_kernel
<
sample_type
>>
(
m
,
"_radial_basis_kernel"
)
.
def
(
"__repr__"
,
radial_basis_kernel__repr__
)
.
def_property_readonly
(
"gamma"
,
[](
const
radial_basis_kernel
<
sample_type
>&
k
){
return
k
.
gamma
;
});
py
::
class_
<
linear_kernel
<
sample_type
>>
(
m
,
"_linear_kernel"
)
.
def
(
"__repr__"
,
linear_kernel__repr__
);
add_df
<
radial_basis_kernel
<
sample_type
>
>
(
m
,
"_decision_function_radial_basis"
);
add_df
<
sparse_radial_basis_kernel
<
sparse_vect
>
>
(
m
,
"_decision_function_sparse_radial_basis"
);
...
...
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