mirror of
https://github.com/sockspls/badfish
synced 2025-05-02 09:39:36 +00:00
35 lines
960 B
C++
35 lines
960 B
C++
// NNUE評価関数で用いる入力特徴量とネットワーク構造の定義
|
|
|
|
#include "../features/feature_set.h"
|
|
#include "../features/half_kp.h"
|
|
|
|
#include "../layers/input_slice.h"
|
|
#include "../layers/affine_transform.h"
|
|
#include "../layers/clipped_relu.h"
|
|
|
|
namespace Eval {
|
|
|
|
namespace NNUE {
|
|
|
|
// 評価関数で用いる入力特徴量
|
|
using RawFeatures = Features::FeatureSet<
|
|
Features::HalfKP<Features::Side::kFriend>>;
|
|
|
|
// 変換後の入力特徴量の次元数
|
|
constexpr IndexType kTransformedFeatureDimensions = 256;
|
|
|
|
namespace Layers {
|
|
|
|
// ネットワーク構造の定義
|
|
using InputLayer = InputSlice<kTransformedFeatureDimensions * 2>;
|
|
using HiddenLayer1 = ClippedReLU<AffineTransform<InputLayer, 32>>;
|
|
using HiddenLayer2 = ClippedReLU<AffineTransform<HiddenLayer1, 32>>;
|
|
using OutputLayer = AffineTransform<HiddenLayer2, 1>;
|
|
|
|
} // namespace Layers
|
|
|
|
using Network = Layers::OutputLayer;
|
|
|
|
} // namespace NNUE
|
|
|
|
} // namespace Eval
|