NumberDemoAgent-AgentStep.cs

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;
		}
	}