Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
bob
bob.bio.base
Commits
caf1f710
Commit
caf1f710
authored
Nov 24, 2020
by
Yannick DAYER
Browse files
Added all_samples to csv dataset classes
parent
20bf6fe4
Changes
1
Hide whitespace changes
Inline
Side-by-side
bob/bio/base/database/csv_dataset.py
View file @
caf1f710
...
...
@@ -326,6 +326,26 @@ class CSVDatasetDevEval:
group
=
group
,
purpose
=
"probe"
,
group_by_subject
=
False
)
def
all_samples
(
self
,
groups
=
None
):
"""
Reads and returns all the samples in `groups`.
Parameters
----------
groups: list or None
Groups to consider, or all groups if `None` is given.
"""
# Get train samples (background_model_samples returns a list of samples)
samples
=
self
.
background_model_samples
()
# Get enroll and probe samples
groups
=
[
"dev"
,
"eval"
]
if
not
groups
else
groups
for
group
in
groups
:
for
purpose
in
(
"enroll"
,
"probe"
):
label
=
f
"
{
group
}
_
{
purpose
}
_csv"
samples
.
append
(
self
.
csv_to_sample_loader
(
self
.
__dict__
[
label
]))
return
samples
class
CSVDatasetCrossValidation
:
"""
...
...
@@ -456,6 +476,26 @@ class CSVDatasetCrossValidation:
def
probes
(
self
,
group
=
"dev"
):
return
self
.
_load_from_cache
(
"dev_probe_csv"
)
def
all_samples
(
self
,
groups
=
None
):
"""
Reads and returns all the samples in `groups`.
Parameters
----------
groups: list or None
Groups to consider, or all groups if `None` is given.
"""
# Get train samples (background_model_samples returns a list of samples)
samples
=
self
.
background_model_samples
()
# Get enroll and probe samples
groups
=
[
"dev"
,
"eval"
]
if
not
groups
else
groups
for
group
in
groups
:
for
purpose
in
(
"enroll"
,
"probe"
):
label
=
f
"
{
group
}
_
{
purpose
}
_csv"
samples
.
append
(
self
.
csv_to_sample_loader
(
self
.
__dict__
[
label
]))
return
samples
def
group_samples_by_subject
(
samples
):
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new 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