| 303 | } |
| 304 | |
| 305 | TFullModel SimpleDeepTreeModel(size_t treeDepth) { |
| 306 | TFullModel model; |
| 307 | TModelTrees* trees = model.ModelTrees.GetMutable(); |
| 308 | for (size_t featureIndex : xrange(treeDepth)) { |
| 309 | const auto feature = TFloatFeature(false, featureIndex, featureIndex, {0.5f}, ""); |
| 310 | trees->AddFloatFeature(feature); |
| 311 | } |
| 312 | for (size_t val : xrange(1 << treeDepth)) { |
| 313 | trees->AddLeafValue(val); |
| 314 | } |
| 315 | TVector<int> tree = xrange(treeDepth); |
| 316 | trees->AddBinTree(tree); |
| 317 | model.UpdateDynamicData(); |
| 318 | return model; |
| 319 | } |
| 320 | |
| 321 | TFullModel SimpleAsymmetricModel() { |
| 322 | TVector<TFloatFeature> floatFeatures { |
no test coverage detected