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> ...@@ -142,7 +142,7 @@ TrainVectorBase<TInputValue, TOutputValue>
std::string tmpKey = "feat." + key.substr( 0, static_cast<unsigned long>( end - key.begin() ) ); std::string tmpKey = "feat." + key.substr( 0, static_cast<unsigned long>( end - key.begin() ) );
this->AddChoice( tmpKey, item ); 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() ) ); std::string tmpKey = "cfield." + key.substr( 0, static_cast<unsigned long>( end - key.begin() ) );
this->AddChoice( tmpKey, item ); 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