Function SmallFloatPool

catboost/libs/model/ut/leaf_weights_ut.cpp:27–69

};

static TDataProviderPtr SmallFloatPool(EWeightsMode addWeights, ETargetDimMode multiclass) {
    return CreateDataProvider(
        [&] (IRawFeaturesOrderDataVisitor* visitor) {
            TDataMetaInfo metaInfo;
            metaInfo.TargetType = multiclass ? ERawTargetType::String : ERawTargetType::Float;
            metaInfo.TargetCount = 1;
            metaInfo.HasWeights = addWeights;
            metaInfo.FeaturesLayout = MakeIntrusive<TFeaturesLayout>(
                (ui32)3,
                TVector<ui32>{},
                TVector<ui32>{},
                TVector<ui32>{},
                TVector<TString>{});

            visitor->Start(metaInfo, 3, EObjectsOrder::Undefined, {});

            visitor->AddFloatFeature(
                0,
                MakeIntrusive<TTypeCastArrayHolder<float, float>>(TVector<float>{+0.5f, +1.5f, -2.5f})
            );
            visitor->AddFloatFeature(
                1,
                MakeIntrusive<TTypeCastArrayHolder<float, float>>(TVector<float>{+0.7f, +6.4f, +2.4f})
            );
            visitor->AddFloatFeature(
                2,
                MakeIntrusive<TTypeCastArrayHolder<float, float>>(TVector<float>{-2.0f, -1.0f, +6.0f})
            );

            if (multiclass) {
                visitor->AddTarget(TVector<TString>{"1", "0", "2"});
            } else {
                visitor->AddTarget(
                    MakeIntrusive<TTypeCastArrayHolder<float, float>>(TVector<float>{1.0f, 0.0f, 0.2f})
                );
            }
            if (addWeights) {
                visitor->AddWeights({1.0f, 2.0f, 0.5f});
            }
            visitor->Finish();
        }
    );
}

static TFullModel TrainModelOnPool(TDataProviderPtr pool, ETargetDimMode multiclass) {
    TFullModel model;