Cli Interface
The cli Interface can be accessed globaly after running julia --project build/build.jl install
.
~/.julia/bin/mlnanoshaper train [options] [flags]
Train a model.
Intro
Train a model that can reconstruct a protein surface using Machine Learning. Default value of parameters are specified in the param/param.toml
file. In order to override the param, you can use the differents options.
Options
--nb-epoch <Int>
: the number of epoch to compute.--model, -m <String>
: the model name. Can be anakin.--nb-data-points <Int>
: the number of proteins in the dataset to use--name, -n <String>
: name of the training run--cutoff-radius, -c <Float32>
: the cutoff_radius used in training--ref-distance <Float32>
: the reference distane (in A) used to rescale distance to surface in loss--learning-rate, -l <Float64>
: the learning rate use by the model in training.--loss <String>
: the loss function, one of "categorical" or "continuous".
Flags
--gpu, -g
: should we do the training on the gpu, does nothing currently.