Skip to content

Commit 45160f6

Browse files
committed
added documentation
1 parent 67a989a commit 45160f6

File tree

3 files changed

+13
-14
lines changed

3 files changed

+13
-14
lines changed

FastTimeSeries/src/container.jl

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,3 @@
1-
#= TODO: loadfile
2-
3-
elseif endswith(file, ".ts")
4-
return _ts2df(file)
5-
6-
=#
7-
81
Datasets.loadfile(file::String, ::Val{:ts}) = _ts2df(file)
92

103
#TimeSeriesDataset

FastTimeSeries/src/models/InceptionTime.jl

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
11
"""
22
InceptionModule(ni::Int, nf::Int, ks::Int = 40, bottleneck::Bool = true)
33
4-
TBW
4+
An InceptionModule consists of an (optional) bottleneck, followed by
5+
3 conv1d layers.
56
"""
67
function InceptionModule(ni::Int, nf::Int, kernel_size::Int = 40, bottleneck::Bool = true)
78
ks = [kernel_size ÷ (2^i) for i in 0:2]
@@ -23,7 +24,8 @@ end
2324
"""
2425
InceptionBlock(ni::Int, nf::Int = 32, residual::Bool = true, depth::Int = 6)
2526
26-
TBW
27+
An InceptionBlock consists of variable number of InceptionModule depending on the depth.
28+
Optionally residual.
2729
"""
2830
function InceptionBlock(ni::Int, nf::Int = 32, residual::Bool = true, depth::Int = 6)
2931
inception = []
@@ -69,9 +71,16 @@ changedims(X) = permutedims(X, (2, 1, 3))
6971
"""
7072
InceptionTime(c_in::Int, c_out::Int, seq_len = nothing, nf::Int = 32)
7173
72-
TBW
74+
A Julia Implemention of the InceptionTime model.
75+
From https://arxiv.org/abs/1909.04939
76+
77+
## Arguments.
78+
79+
- `c_in` : The number of input channels.
80+
- `c_out`: The number of output classes.
81+
- `nf` : The number of "hidden channels" to use.
7382
"""
74-
function InceptionTime(c_in::Int, c_out::Int, seq_len = nothing, nf::Int = 32)
83+
function InceptionTime(c_in::Int, c_out::Int, nf::Int = 32)
7584
inceptionblock = InceptionBlock(c_in, nf)
7685
gap = GAP1d(1)
7786
fc = Dense(nf * 4, c_out)

FastTimeSeries/src/recipes.jl

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,9 +18,7 @@ function Datasets.recipeblocks(recipe::TimeSeriesDatasetRecipe)
1818
return Tuple{TimeSeriesRow, Continuous}
1919
end
2020
end
21-
# Datasets.recipeblocks(::Type{TimeSeriesDatasetRecipe}) = Tuple{TimeSeriesRow, Label}
2221

23-
#TODO: Add Check if test_file is nothing.
2422
function Datasets.loadrecipe(recipe::TimeSeriesDatasetRecipe, path)
2523
path = convert(String, path)
2624
datasetpath_train = joinpath(path, recipe.train_file)
@@ -47,7 +45,6 @@ function Datasets.loadrecipe(recipe::TimeSeriesDatasetRecipe, path)
4745
end
4846

4947
# Registering recipes
50-
5148
const RECIPES = Dict{String,Vector{Datasets.DatasetRecipe}}(
5249
"ecg5000" => [
5350
TimeSeriesDatasetRecipe(train_file="ECG5000_TRAIN.ts", test_file="ECG5000_TEST.ts")

0 commit comments

Comments
 (0)