| 122 | } |
| 123 | |
| 124 | void OutputModelOnnx( |
| 125 | const TFullModel& model, |
| 126 | const TString& modelFile, |
| 127 | const TString& userParametersJson) { |
| 128 | |
| 129 | /* TODO(akhropov): the problem with OneHotFeatures is that raw 'float' values |
| 130 | * could be interpreted as nans so that equality comparison won't work for such splits |
| 131 | */ |
| 132 | TOFStream out(modelFile); |
| 133 | SerializeFullModelToOnnxStream(model, userParametersJson, &out); |
| 134 | } |
| 135 | |
| 136 | void ExportModel( |
| 137 | const TFullModel& model, |
no test coverage detected