|
1 | 1 | module MLJModelInterface
|
2 | 2 |
|
3 | 3 | # ------------------------------------------------------------------------
|
4 |
| -# Dependency (note that ScientificTypes itself does not have dependencies) |
| 4 | +# Dependency (ScientificTypes itself does not have dependencies) |
5 | 5 | using ScientificTypes
|
6 | 6 |
|
7 | 7 | # ------------------------------------------------------------------------
|
8 | 8 | # exports
|
9 | 9 |
|
10 |
| -# types |
| 10 | +# mode |
11 | 11 | export LightInterface, FullInterface
|
| 12 | + |
| 13 | +# MLJ model hierarchy |
12 | 14 | export MLJType, Model, Supervised, Unsupervised,
|
13 | 15 | Probabilistic, Deterministic, Interval, Static,
|
14 | 16 | UnivariateFinite
|
15 | 17 |
|
16 |
| -# rexport types from ScientificTypes |
| 18 | +# reexport types from ScientificTypes |
17 | 19 | export Scientific, Found, Unknown, Known, Finite, Infinite,
|
18 | 20 | OrderedFactor, Multiclass, Count, Continuous, Textual,
|
19 | 21 | Binary, ColorImage, GrayImage, Table
|
20 | 22 |
|
21 |
| -# constructor + metadata |
| 23 | +# model constructor + metadata |
22 | 24 | export @mlj_model, metadata_pkg, metadata_model, metadata_measure
|
23 |
| -# api |
| 25 | + |
| 26 | +# model api |
24 | 27 | export fit, update, update_data, transform, inverse_transform,
|
25 | 28 | fitted_params, predict, predict_mode, predict_mean, predict_median,
|
26 | 29 | evaluate, clean!
|
27 |
| -# traits |
| 30 | + |
| 31 | +# model traits |
28 | 32 | export input_scitype, output_scitype, target_scitype,
|
29 | 33 | is_pure_julia, package_name, package_license,
|
30 | 34 | load_path, package_uuid, package_url,
|
|
0 commit comments