Skip to content

Rename sum_hess field to weighted_data_count #662

Description

@david-cortes

When building a treelite model, there is a method data_count which takes as input:

data_count – Number of data points

But if a given tree was fitted to data with sample weights, one would want it to instead represent the sum of weights among the observations, which will not necessarily be integral.

There is a sum_hess which takes float, but it's not exactly the same thing.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions