62 lines
1.5 KiB
JavaScript
62 lines
1.5 KiB
JavaScript
const DEFAULT_EPSILON = 0.01;
|
|
|
|
function create_key(steps) {
|
|
return steps.join('');
|
|
}
|
|
|
|
export default class Daemon {
|
|
humanCount;
|
|
robotCount;
|
|
epsilon;
|
|
weights = {};
|
|
|
|
constructor(humanCount, robotCount, epsilon = DEFAULT_EPSILON) {
|
|
this.humanCount = humanCount;
|
|
this.robotCount = robotCount;
|
|
this.epsilon = epsilon;
|
|
}
|
|
|
|
get power() {
|
|
return this.humanCount + this.robotCount;
|
|
}
|
|
|
|
predict(journal) {
|
|
const steps = this._getStepSlice(journal);
|
|
const w0 = this._getWeight([...steps, 0]);
|
|
const w1 = this._getWeight([...steps, 1]);
|
|
|
|
return w1 > w0 ? 1 : 0;
|
|
}
|
|
|
|
adjust(journal, humanValue) {
|
|
const steps = this._getStepSlice(journal);
|
|
const adjustmentWeight = this._getAdjustmentWeight(journal.length);
|
|
this._adjustWeight([...steps, humanValue], adjustmentWeight);
|
|
}
|
|
|
|
_getStepSlice(journal) {
|
|
return journal.getLastMovements(this.humanCount, this.robotCount);
|
|
}
|
|
|
|
_getAdjustmentWeight(stepNumber) {
|
|
return Math.pow(1 + this.epsilon, stepNumber);
|
|
}
|
|
|
|
_getWeight(steps) {
|
|
const key = create_key(steps);
|
|
const weight = this.weights[key];
|
|
return weight === undefined ? 0 : weight;
|
|
}
|
|
|
|
_setWeight(steps, value) {
|
|
const key = create_key(steps);
|
|
this.weights[key] = value;
|
|
}
|
|
|
|
_adjustWeight(steps, weight) {
|
|
const currentWeight = this._getWeight(steps);
|
|
const newWeight = currentWeight + weight;
|
|
this._setWeight(steps, newWeight);
|
|
}
|
|
}
|