Skip to content

Instantly share code, notes, and snippets.

@unity3dcollege
Created November 1, 2017 22:23
Show Gist options
  • Save unity3dcollege/d2d64a7264dbd29a48e80f591a1e214b to your computer and use it in GitHub Desktop.
Save unity3dcollege/d2d64a7264dbd29a48e80f591a1e214b to your computer and use it in GitHub Desktop.
public override void AgentStep(float[] action)
{
if (text != null)
text.text = string.Format("C:{0} / T:{1} [{2}]", currentNumber, targetNumber, solved);
switch ((int)action[0])
{
case 0:
currentNumber -= 0.01f;
break;
case 1:
currentNumber += 0.01f;
break;
default:
return;
}
cube.position = new Vector3(currentNumber * 5f, 0f, 0f);
if (currentNumber < -1.2f || currentNumber > 1.2f)
{
reward = -1f;
done = true;
return;
}
float difference = Mathf.Abs(targetNumber - currentNumber);
if (difference <= 0.01f)
{
solved++;
reward = 1;
done = true;
return;
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment