simple chgs
This commit is contained in:
parent
f7b2281510
commit
604568b512
2
Makefile
2
Makefile
@ -1,6 +1,6 @@
|
|||||||
export CC = gcc
|
export CC = gcc
|
||||||
export CXX = g++
|
export CXX = g++
|
||||||
export CFLAGS = -Wall -O3 -msse2 -Wno-unknown-pragmas -fopenmp
|
export CFLAGS = -Wall -O3 -msse2 -Wno-unknown-pragmas -fopenmp
|
||||||
|
|
||||||
# specify tensor path
|
# specify tensor path
|
||||||
BIN = xgboost
|
BIN = xgboost
|
||||||
|
|||||||
@ -104,7 +104,7 @@ namespace xgboost{
|
|||||||
|
|
||||||
namespace xgboost{
|
namespace xgboost{
|
||||||
namespace regrank{
|
namespace regrank{
|
||||||
IObjFunction* CreateObjFunction( const char *name ){
|
inline IObjFunction* CreateObjFunction( const char *name ){
|
||||||
if( !strcmp("reg", name ) ) return new RegressionObj();
|
if( !strcmp("reg", name ) ) return new RegressionObj();
|
||||||
if( !strcmp("rank:pairwise", name ) ) return new PairwiseRankObj();
|
if( !strcmp("rank:pairwise", name ) ) return new PairwiseRankObj();
|
||||||
if( !strcmp("rank:softmax", name ) ) return new SoftmaxRankObj();
|
if( !strcmp("rank:softmax", name ) ) return new SoftmaxRankObj();
|
||||||
|
|||||||
@ -339,7 +339,7 @@ namespace xgboost{
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
virtual inline void GetLambda(const std::vector<float> &preds,
|
virtual void GetLambda(const std::vector<float> &preds,
|
||||||
const std::vector<float> &labels,
|
const std::vector<float> &labels,
|
||||||
const std::vector<unsigned> &group_index,
|
const std::vector<unsigned> &group_index,
|
||||||
const std::vector< std::pair<int, int> > &pairs, std::vector<float> lambda, int group) = 0;
|
const std::vector< std::pair<int, int> > &pairs, std::vector<float> lambda, int group) = 0;
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user