MLNanoShaper

accumulator(processing,logger)

A processing logger that transform logger on multiple batches Ca be used to liss numerical data, for logging to TensorBoardLogger.

AuxiliaryParameters

The variables that do not influence the outome of the training run. This include the nb_epoch.

abstract type LossType end

LossType is an interface for defining loss functions.

Implementation

• getlossfn(::LossType)::Function : the associated loss function
• metrictype(::Type{<:LossType)::Type : the type of metrics returned by the loss function
• getlosstype(::StaticSymbol)::LossType : the function generating the losstype

Training information used in model training.

Fields

• atoms: the set of atoms used as model input
• skin : the Surface generated by Nanoshaper

TrainingParameters

The training parameters used in the model training. Default values are in the param file. The training is deterministric. Theses values are hased to determine a training run

_train(training_parameters::TrainingParameters, directories::AuxiliaryParameters)

train the model given TrainingParameters and AuxiliaryParameters.

train((train_data,test_data),training_states; nb_epoch)

train the model on the data with nb_epoch

categorical_loss(model, ps, st, (; point, atoms, d_real))

The loss function used by in training. Return the KL divergence between true probability and empirical probability Return the error with the espected distance as a metric.

continus_loss(model, ps, st, (; point, atoms, d_real))

The loss function used by in training. compare the predicted (square) distance with $\frac{1 + anh(d)}{2}$ Return the error with the espected distance as a metric.

generate_data()

generate data from the parameters files in param/ by downloading the pdb files and running Nanoshaper.

generate_data_points(
    preprocessing::Lux.AbstractExplicitLayer, points::AbstractVector{<:Point3},
    (; atoms, skin)::TreeTrainingData{Float32}, (; ref_distance)::TrainingParameters)

generate the data_points for a set of positions points on one protein.

implicit_surface(atoms::AnnotedKDTree{Sphere{T}, :center, Point3{T}},
    model::Lux.StatefulLuxLayer, (;
        cutoff_radius, step)) where {T}

Create a mesh form the isosurface of function `pos -> model(atoms,pos)` using marching cubes algorithm and using step size `step`.

load_data_pdb(T, name::String)

Load a TrainingData{T} from current directory. You should have a pdb and an off file with name name in current directory.

load_data_pqr(T, name::String)

Load a TrainingData{T} from current directory. You should have a pdb and an off file with name name in current directory.

state

The global state manipulated by the c interface. To use, you must first load the weights using load_weights and the input atoms using load_atoms. Then you can call eval_model to get the field on a certain point.

AnnotedKDTree(data::StructVector,property::StaticSymbol)

Fields

• data::StructVector
• tree::KDTree

ConcatenatedBatch

Represent a vector of arrays of sizes (a..., bn) where bn is the variable dimension of the batch. You can access view of individual arrays with get_slice.

ModelInput

input of the model

Fields

• point::Point3, the position of the input
• atoms::StructVector{Sphere}, the atoms in the neighboord

batched_sum(b::AbstractMatrix,nb_elements::AbstractVector)

compute the sum of a Concatenated batch with ndim = 2. The first dim is the feature dimension. The second dim is the the batch dim.

Given b of size (n,m) and nb_elements of size (k,), the output has size (n,k).

distance(x::GeometryBasics.Mesh, y::KDTree)

Return the Hausdorff distance betwen the mesh coordinates

eval_model(x::Float32,y::Float32,z::Float32)::Float32

evaluate the model at coordinates x y z.

evaluate_model

light_angular_dense(; categorical=false, van_der_waals_channel=false, kargs...)

`light_angular_dense` is a function that generate a lux model.

load_atoms(start::Ptr{CSphere},length::Cint)::Cint

Load the atoms into the julia model. Start is a pointer to the start of the array of CSphere and length is the length of the array

Return an error status:

• 0: OK
• 1: data could not be read
• 2: unknow error

load_model(path::String)::Cint

Load the model from a MLNanoShaperRunner.SerializedModel serialized state at absolute path path.

Return an error status:

• 0: OK
• 1: file not found
• 2: file could not be deserialized properly
• 3: unknow error

medium_angular_dense(; categorical=false, van_der_waals_channel=false, kargs...)

`medium_angular_dense` is a function that generate a lux model.

signed_distance(p::Point3, mesh::RegionMesh)::Number

returns the signed distance between point p and the mesh

tiny_angular_dense(; categorical=false, van_der_waals_channel=false, kargs...)

`tiny_angular_dense` is a function that generate a lux model.