mirror of
https://github.com/vcmi/vcmi.git
synced 2025-03-25 21:38:59 +02:00
[programming challenge, SSN] enemy's army is no longer part of ANN's input
This commit is contained in:
parent
3d16f0a081
commit
4a7891139e
@ -91,6 +91,7 @@
|
||||
<Optimization>Disabled</Optimization>
|
||||
<PrecompiledHeader>Use</PrecompiledHeader>
|
||||
<PrecompiledHeaderFile>StdInc.h</PrecompiledHeaderFile>
|
||||
<AdditionalOptions>/Zm150 %(AdditionalOptions)</AdditionalOptions>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
|
@ -364,49 +364,47 @@ double rateArt(const DuelParameters dp, CArtifactInstance * inst)
|
||||
}
|
||||
|
||||
|
||||
const unsigned int num_input = 27;
|
||||
const unsigned int num_input = 18;
|
||||
|
||||
double * genSSNinput(const DuelParameters & dp, CArtifactInstance * art)
|
||||
double * genSSNinput(const DuelParameters::SideSettings & dp, CArtifactInstance * art, si32 bfieldType, si32 terType)
|
||||
{
|
||||
double * ret = new double[num_input];
|
||||
double * cur = ret;
|
||||
|
||||
//general description
|
||||
|
||||
*(cur++) = dp.bfieldType/30.0;
|
||||
*(cur++) = dp.terType/12.0;
|
||||
*(cur++) = bfieldType/30.0;
|
||||
*(cur++) = terType/12.0;
|
||||
|
||||
//creature & hero description
|
||||
|
||||
for(int i=0; i<2; ++i)
|
||||
|
||||
*(cur++) = dp.heroId/200.0;
|
||||
for(int k=0; k<4; ++k)
|
||||
*(cur++) = dp.heroPrimSkills[k]/20.0;
|
||||
|
||||
//weighted average of statistics
|
||||
auto avg = [&](std::function<int(CCreature *)> getter) -> double
|
||||
{
|
||||
auto & side = dp.sides[0];
|
||||
*(cur++) = side.heroId/200.0;
|
||||
for(int k=0; k<4; ++k)
|
||||
*(cur++) = side.heroPrimSkills[k]/20.0;
|
||||
|
||||
//weighted average of statistics
|
||||
auto avg = [&](std::function<int(CCreature *)> getter) -> double
|
||||
double ret = 0.0;
|
||||
int div = 0;
|
||||
for(int i=0; i<7; ++i)
|
||||
{
|
||||
double ret = 0.0;
|
||||
int div = 0;
|
||||
for(int i=0; i<7; ++i)
|
||||
auto & cstack = dp.stacks[i];
|
||||
if(cstack.count > 0)
|
||||
{
|
||||
auto & cstack = side.stacks[i];
|
||||
if(cstack.count > 0)
|
||||
{
|
||||
ret += getter(VLC->creh->creatures[cstack.type]) * cstack.count;
|
||||
div+=cstack.count;
|
||||
}
|
||||
ret += getter(VLC->creh->creatures[cstack.type]) * cstack.count;
|
||||
div+=cstack.count;
|
||||
}
|
||||
return ret/div;
|
||||
};
|
||||
}
|
||||
return ret/div;
|
||||
};
|
||||
|
||||
*(cur++) = avg([](CCreature * c){return c->attack;})/50.0;
|
||||
*(cur++) = avg([](CCreature * c){return c->defence;})/50.0;
|
||||
*(cur++) = avg([](CCreature * c){return c->speed;})/15.0;
|
||||
*(cur++) = avg([](CCreature * c){return c->hitPoints;})/1000.0;
|
||||
}
|
||||
|
||||
*(cur++) = avg([](CCreature * c){return c->attack;})/50.0;
|
||||
*(cur++) = avg([](CCreature * c){return c->defence;})/50.0;
|
||||
*(cur++) = avg([](CCreature * c){return c->speed;})/15.0;
|
||||
*(cur++) = avg([](CCreature * c){return c->hitPoints;})/1000.0;
|
||||
|
||||
//bonus description
|
||||
auto & blist = art->getBonusList();
|
||||
@ -425,7 +423,7 @@ double * genSSNinput(const DuelParameters & dp, CArtifactInstance * art)
|
||||
//returns how good the artifact is for the neural network
|
||||
double runSSN(FANN::neural_net & net, const DuelParameters dp, CArtifactInstance * inst)
|
||||
{
|
||||
double * input = genSSNinput(dp, inst);
|
||||
double * input = genSSNinput(dp.sides[0], inst, dp.bfieldType, dp.terType);
|
||||
double * out = net.run(input);
|
||||
double ret = *out;
|
||||
free(out);
|
||||
@ -450,9 +448,10 @@ void learnSSN(FANN::neural_net & net, const std::vector<Example> & input)
|
||||
double ** outputs = new double *[input.size()];
|
||||
for(int i=0; i<input.size(); ++i)
|
||||
{
|
||||
inputs[i] = genSSNinput(input[i].dp, input[i].art);
|
||||
const auto & ci = input[i];
|
||||
inputs[i] = genSSNinput(ci.dp.sides[0], ci.art, ci.dp.bfieldType, ci.dp.terType);
|
||||
outputs[i] = new double;
|
||||
*(outputs[i]) = input[i].value;
|
||||
*(outputs[i]) = ci.value;
|
||||
}
|
||||
td.set_train_data(input.size(), num_input, inputs, 1, outputs);
|
||||
net.set_callback(ANNCallback, NULL);
|
||||
@ -738,7 +737,7 @@ int main(int argc, char **argv)
|
||||
VLC->init();
|
||||
|
||||
|
||||
buildLearningSet();
|
||||
//buildLearningSet();
|
||||
|
||||
SSNRun();
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user