Skip to content

Commit 18cb1da

Browse files
committed
added multivariate clf dataset
1 parent 52a1932 commit 18cb1da

File tree

2 files changed

+12
-0
lines changed

2 files changed

+12
-0
lines changed

FastTimeSeries/src/recipes.jl

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,16 @@ const RECIPES = Dict{String,Vector{Datasets.DatasetRecipe}}(
3636
"ecg5000" => [
3737
TimeSeriesDatasetRecipe(train_file="ECG5000_TRAIN.ts", test_file="ECG5000_TEST.ts")
3838
],
39+
"atrial" => [
40+
TimeSeriesDatasetRecipe(train_file="AtrialFibrillation_TRAIN.ts", test_file="AtrialFibrillation_TEST.ts")
41+
],
42+
"natops" => [
43+
TimeSeriesDatasetRecipe(train_file="NATOPS_TEST.ts", test_file="NATOPS_TRAIN.ts")
44+
]
45+
#! TODO.
46+
# "appliances_energy" => [
47+
# TimeSeriesDatasetRecipe(train_file="AppliancesEnergy_TRAIN.ts", test_file="AppliancesEnergy_TEST.ts")
48+
# ]
3949
)
4050

4151
function _registerrecipes()

src/datasets/fastaidatasets.jl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@ const DESCRIPTIONS = Dict(
4646
"food-101" => "101 food categories, with 101,000 images; 250 test images and 750 training images per class. The training images were not cleaned. All images were rescaled to have a maximum side length of 512 pixels.",
4747
"ECG5000" => "The original dataset for \"ECG5000\" is a 20-hour long ECG downloaded from Physionet. The name is BIDMC Congestive Heart Failure Database(chfdb) and it is record \"chf07\".",
4848
"AtrialFibrillation" => "This is a physionet dataset of two-channel ECG recordings has been created from data used in the Computers in Cardiology Challenge 2004, an open competition with the goal of developing automated methods for predicting spontaneous termination of atrial fibrillation (AF).",
49+
"NATOPS" => "The data is generated by sensors on the hands, elbows, wrists and thumbs. The data are the x,y,z coordinates for each of the eight locations. "
4950
)
5051

5152
const DATASETCONFIGS = [
@@ -206,6 +207,7 @@ const DATASETCONFIGS = [
206207
# timeseries
207208
TSClassificationDataset("ECG5000", "41f6de20ac895e9ce31753860995518951f1ed42a405d0e51c909d27e3b3c5a4", description = DESCRIPTIONS["ECG5000"] ,datadepname="ecg5000", size="10MB" ),
208209
TSClassificationDataset("AtrialFibrillation", "218abad67d58190a6daa1a27f4bd58ace6e18f80fb59fb2c7385f0d2d4b411a2", description = DESCRIPTIONS["AtrialFibrillation"], datadepname = "atrial", size = "226KB"),
210+
TSClassificationDataset("NATOPS", "57a8debeedadad7764bfa9c87b4300bd64a999ef95a98a6ee07a830c41de4aa1", description = DESCRIPTIONS["NATOPS"], datadepname = "natops", size = "5.1MB"),
209211

210212
]
211213

0 commit comments

Comments
 (0)