"src/vscode:/vscode.git/clone" did not exist on "4f28233b8ff705cbf14140ebd71c4e5cc1ab2c48"
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 { ...@@ -141,7 +141,7 @@ class GBDT : public GBDTBase {
/*! /*!
* \brief Training logic * \brief Training logic
* \param gradients nullptr for using default objective, otherwise use self-defined boosting * \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 * \return True if cannot train any more
*/ */
bool TrainOneIter(const score_t* gradients, const score_t* hessians) override; bool TrainOneIter(const score_t* gradients, const score_t* hessians) override;
......
...@@ -138,11 +138,11 @@ class GPUTreeLearner: public SerialTreeLearner { ...@@ -138,11 +138,11 @@ class GPUTreeLearner: public SerialTreeLearner {
* Set to nullptr to skip copy to GPU. * Set to nullptr to skip copy to GPU.
* \param num_data Number of data examples to be included in histogram * \param num_data Number of data examples to be included in histogram
* \param gradients Array of gradients for all examples. * \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, * \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. * 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, * \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. * Set hessians to nullptr to skip copy to GPU.
* \return true if GPU kernel is launched, false if GPU is not used * \return true if GPU kernel is launched, false if GPU is not used
*/ */
bool ConstructGPUHistogramsAsync( 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