Unverified Commit fcd24535 authored by Nikita Titov's avatar Nikita Titov Committed by GitHub
Browse files

fix param name (#4253)

* fix param name

* Update gpu_tree_learner.h

* Update gbdt.h
parent ec874325
......@@ -141,7 +141,7 @@ class GBDT : public GBDTBase {
/*!
* \brief Training logic
* \param gradients nullptr for using default objective, otherwise use self-defined boosting
* \param Hessians nullptr for using default objective, otherwise use self-defined boosting
* \param hessians nullptr for using default objective, otherwise use self-defined boosting
* \return True if cannot train any more
*/
bool TrainOneIter(const score_t* gradients, const score_t* hessians) override;
......
......@@ -138,11 +138,11 @@ class GPUTreeLearner: public SerialTreeLearner {
* Set to nullptr to skip copy to GPU.
* \param num_data Number of data examples to be included in histogram
* \param gradients Array of gradients for all examples.
* \param Hessians Array of Hessians for all examples.
* \param hessians Array of Hessians for all examples.
* \param ordered_gradients Ordered gradients will be generated and copied to GPU when gradients is not nullptr,
* Set gradients to nullptr to skip copy to GPU.
* \param ordered_hessians Ordered Hessians will be generated and copied to GPU when Hessians is not nullptr,
* Set Hessians to nullptr to skip copy to GPU.
* \param ordered_hessians Ordered Hessians will be generated and copied to GPU when hessians is not nullptr,
* Set hessians to nullptr to skip copy to GPU.
* \return true if GPU kernel is launched, false if GPU is not used
*/
bool ConstructGPUHistogramsAsync(
......
Markdown is supported
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