1
0
Fork 0
mirror of https://github.com/sockspls/badfish synced 2025-07-11 19:49:14 +00:00
BadFish/src/eval/nnue/evaluate_nnue_learner.h

46 lines
1.2 KiB
C++

// NNUE評価関数の学習で用いるインターフェイス
#ifndef _EVALUATE_NNUE_LEARNER_H_
#define _EVALUATE_NNUE_LEARNER_H_
#if defined(EVAL_LEARN) && defined(EVAL_NNUE)
#include "../../learn/learn.h"
namespace Eval {
namespace NNUE {
// 学習の初期化を行う
void InitializeTraining(double eta1, uint64_t eta1_epoch,
double eta2, uint64_t eta2_epoch, double eta3);
// ミニバッチのサンプル数を設定する
void SetBatchSize(uint64_t size);
// 学習率のスケールを設定する
void SetGlobalLearningRateScale(double scale);
// ハイパーパラメータなどのオプションを設定する
void SetOptions(const std::string& options);
// 学習用評価関数パラメータをファイルから読み直す
void RestoreParameters(const std::string& dir_name);
// 学習データを1サンプル追加する
void AddExample(Position& pos, Color rootColor,
const Learner::PackedSfenValue& psv, double weight);
// 評価関数パラメータを更新する
void UpdateParameters(uint64_t epoch);
// 学習に問題が生じていないかチェックする
void CheckHealth();
} // namespace NNUE
} // namespace Eval
#endif // defined(EVAL_LEARN) && defined(EVAL_NNUE)
#endif