opelly/Stockfish
1
0
Fork 0
mirror of https://github.com/opelly27/Stockfish.git synced 2026-05-20 15:37:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add 'validation_count' option for 'learn' that specifies how many positions to use for validation

Browse Source
This commit is contained in:
Tomasz Sobczyk
2020-12-19 23:46:18 +01:00
committed by nodchip
parent a7378f3249
commit f56613ebf6
3 changed files with 44 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/learn/sfen_reader.h
+12 -36
View File
@@ -73,10 +73,10 @@ namespace Learner{
}
// Load the phase for calculation such as mse.
PSVector read_for_mse(uint64_t count)
PSVector read_some(uint64_t count, int eval_limit, bool use_draw_games)
{
PSVector sfen_for_mse;
sfen_for_mse.reserve(count);
PSVector psv;
psv.reserve(count);
for (uint64_t i = 0; i < count; ++i)
{
@@ -84,43 +84,19 @@ namespace Learner{
if (!read_to_thread_buffer(0, ps))
{
std::cout << "ERROR (sfen_reader): Reading failed." << std::endl;
return sfen_for_mse;
return psv;
}
sfen_for_mse.push_back(ps);
if (eval_limit < abs(ps.score))
continue;
if (!use_draw_games && ps.game_result == 0)
continue;
psv.push_back(ps);
}
return sfen_for_mse;
}
PSVector read_validation_set(const std::string& file_name, int eval_limit, bool use_draw_games)
{
PSVector sfen_for_mse;
auto input = open_sfen_input_file(file_name);
while(!input->eof())
{
std::optional<PackedSfenValue> p_opt = input->next();
if (p_opt.has_value())
{
auto& p = *p_opt;
if (eval_limit < abs(p.score))
continue;
if (!use_draw_games && p.game_result == 0)
continue;
sfen_for_mse.push_back(p);
}
else
{
break;
}
}
return sfen_for_mse;
return psv;
}
// [ASYNC] Thread returns one aspect. Otherwise returns false.