aion-server 4.8

package com.aionemu.gameserver.custom.instance.neuralnetwork;

import java.util.ArrayList;
import java.util.List;

import com.aionemu.gameserver.custom.instance.CustomInstanceService;
import com.aionemu.gameserver.utils.ThreadPoolManager;

/**
 * @author Jo
 */
public class PlayerModelController {

	private static final int TRAINING_EPOCHS = 1000;

	public static PlayerModel trainModelForPlayer(int playerId, List<Integer> skillSet) {
		// get input data:
		List<PlayerModelEntry> playerModelEntries = CustomInstanceService.getInstance().getPlayerModelEntries(playerId);

		// specify model:
		List<DataSet> dataSets = new ArrayList<>();

		for (int i = 0; i < playerModelEntries.size(); i++) {
			int previousSkillID = -1;
			if (i > 0)
				previousSkillID = playerModelEntries.get(i - 1).getSkillID();
			dataSets.add(
				new DataSet(playerModelEntries.get(i).toStateInputArray(skillSet, previousSkillID), playerModelEntries.get(i).toActionOutputArray(skillSet)));
		}

		if (dataSets.isEmpty())
			return null;

		PlayerModel model = new PlayerModel(dataSets.get(0).getValues().length, 10, dataSets.get(0).getTargets().length, 1, null, null);

		// train
		ThreadPoolManager.getInstance().executeLongRunning(() -> model.train(dataSets, TRAINING_EPOCHS));
		return model;
	}

	public static List<Integer> getSkillSetForPlayer(int playerId) {
		List<Integer> skillSet = new ArrayList<>();
		for (PlayerModelEntry pme : CustomInstanceService.getInstance().getPlayerModelEntries(playerId))
			if (!skillSet.contains(pme.getSkillID()))
				skillSet.add(pme.getSkillID());

		return skillSet;
	}

	public static int getActionOutput(PlayerModel model, double[] inputArray) {
		List<Double> targetValues = model.getOutputEstimation(inputArray);
		return getMaxIndex(targetValues);
	}

	public static int getMaxIndex(List<Double> values) {
		int i = 0;
		int actionI = 0;
		double max_t = 0;

		for (double t : values) {
			if (t > max_t) {
				max_t = t;
				actionI = i;
			}
			i++;
		}
		return actionI;

	}

}