NeuroTreeModels

Appearance

API ​

Training ​

MLJModelInterface.fit Function
julia
function fit(
    config::NeuroTypes,
    dtrain;
    feature_names,
    target_name,
    weight_name=nothing,
    offset_name=nothing,
    deval=nothing,
    print_every_n=9999,
    verbosity=1,
)

Training function of NeuroTreeModels' internal API.

Arguments

  • config::NeuroTypes

  • dtrain: Must be <:AbstractDataFrame

Keyword arguments

  • feature_names: Required kwarg, a Vector{Symbol} or Vector{String} of the feature names.

  • target_name Required kwarg, a Symbol or String indicating the name of the target variable.

  • weight_name=nothing

  • offset_name=nothing

  • deval=nothing Data for tracking evaluation metric and perform early stopping.

  • print_every_n=9999

  • verbosity=1

source

Inference ​

NeuroTreeModels.infer Function

infer(m::NeuroTreeModel, data)

Return the inference of a NeuroTreeModel over data, where data is AbstractDataFrame.

source