/
MLJFlux.jl
36 lines (30 loc) · 974 Bytes
/
MLJFlux.jl
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
module MLJFlux
export CUDALibs, CPU1
import Flux
using MLJModelInterface
using MLJModelInterface.ScientificTypes
import Base.==
using ProgressMeter
using CategoricalArrays
using Tables
using Statistics
using ColorTypes
using ComputationalResources
include("core.jl")
include("regressor.jl")
include("classifier.jl")
include("image.jl")
include("common.jl")
### Package specific model traits:
MLJModelInterface.metadata_pkg.((NeuralNetworkRegressor,
MultitargetNeuralNetworkRegressor,
NeuralNetworkClassifier,
ImageClassifier),
name="MLJFlux",
uuid="094fc8d1-fd35-5302-93ea-dabda2abf845",
url="https://github.com/alan-turing-institute/MLJFlux.jl",
julia=true,
license="MIT")
export NeuralNetworkRegressor, MultitargetNeuralNetworkRegressor
export NeuralNetworkClassifier, ImageClassifier
end #module