Skip to content

Commit

Permalink
some further move
Browse files Browse the repository at this point in the history
  • Loading branch information
mli committed Feb 4, 2016
1 parent 58d525a commit 23de1ba
Show file tree
Hide file tree
Showing 5 changed files with 4 additions and 3 deletions.
2 changes: 2 additions & 0 deletions src/bcd/bcd_learner.cc
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
#include "loss/bin_class_metric.h"
namespace difacto {

DMLC_REGISTER_PARAMETER(BCDUpdaterParam);

KWArgs BCDLearner::Init(const KWArgs& kwargs) {
auto remain = Learner::Init(kwargs);
// init param
Expand Down
2 changes: 1 addition & 1 deletion src/main.cc
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
#include "difacto/learner.h"
#include "common/arg_parser.h"
#include "dmlc/parameter.h"
#include "data/converter.h"
#include "reader/converter.h"
namespace difacto {
struct DifactoParam : public dmlc::Parameter<DifactoParam> {
/**
Expand Down
2 changes: 1 addition & 1 deletion src/data/converter.h → src/reader/converter.h
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
* Copyright (c) 2015 by Contributors
*/
#ifndef DIFACTO_DATA_CONVERTER_H_
#define DIFACTO_DATA__CONVERTER_H_
#define DIFACTO_DATA_CONVERTER_H_
#include "dmlc/parameter.h"
#include "reader/reader.h"
#include "dmlc/io.h"
Expand Down
File renamed without changes.
1 change: 0 additions & 1 deletion src/updater.cc
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,5 @@
namespace difacto {

DMLC_REGISTER_PARAMETER(SGDUpdaterParam);
DMLC_REGISTER_PARAMETER(BCDUpdaterParam);

} // namespace difacto

0 comments on commit 23de1ba

Please sign in to comment.