ml-agents/unity-environment/Assets/ML-Agents/Examples/3DBall/Scripts/Ball3DDecision.cs

using System.Collections;
using System.Collections.Generic;
using UnityEngine;

public class Ball3DDecision : MonoBehaviour, Decision
{
    public float[] Decide(List<float> state, List<Camera> observation, float reward, bool done, float[] memory)
    {
        if (gameObject.GetComponent<Brain>().brainParameters.actionSpaceType == StateType.continuous)
        {
            
            return new float[2]{ -10*(state[4]-state[2]), -10*(state[2]+state[4])};

        }
        else
        {
            return new float[1]{ 1f };
        }
    }

    public float[] MakeMemory(List<float> state, List<Camera> observation, float reward, bool done, float[] memory)
    {
        return new float[0];
    }
}