Commit d8b54200 authored by Cédric Traizet's avatar Cédric Traizet
Browse files

ENH: allow real label types in TrainVectorBase for regression

No related merge requests found
Showing with 1 addition and 1 deletion
+1 -1
......@@ -142,7 +142,7 @@ TrainVectorBase<TInputValue, TOutputValue>
std::string tmpKey = "feat." + key.substr( 0, static_cast<unsigned long>( end - key.begin() ) );
this->AddChoice( tmpKey, item );
}
if( fieldType == OFTString || fieldType == OFTInteger || fieldType == OFTInteger64 )
if( fieldType == OFTString || fieldType == OFTInteger || fieldType == OFTInteger64 || fieldType == OFTReal )
{
std::string tmpKey = "cfield." + key.substr( 0, static_cast<unsigned long>( end - key.begin() ) );
this->AddChoice( tmpKey, item );
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment