or-tools: Планирование сотрудников Python не дает никаких решений, если num_nurses не равно num_shifts.

Я пытаюсь использовать для ознакомления с google or-tools. Я попробовал упрощенную версию примера Python для планирования сотрудников.

from __future__ import print_function
import sys
from ortools.constraint_solver import pywrapcp


def main():
    # Creates the solver.
    solver = pywrapcp.Solver("employee_scheduling")

    num_nurses = 3
    num_shifts = 3
    num_days = 1

    # [START]
    # Create shift variables.
    shifts = {}

    for j in range(num_nurses):
        for i in range(num_days):
            shifts[(j, i)] = solver.IntVar(
                0, num_shifts - 1, "shifts(%i,%i)" % (j, i))
    shifts_flat = [shifts[(j, i)] for j in range(num_nurses)
                for i in range(num_days)]

    # Create nurse variables.
    nurses = {}

    for j in range(num_shifts):
        for i in range(num_days):
            nurses[(j, i)] = solver.IntVar(
                0, num_nurses - 1, "shift%d day%d" % (j, i))
    # Set relationships between shifts and nurses.
    for day in range(num_days):
        nurses_for_day = [nurses[(j, day)] for j in range(num_shifts)]

        for j in range(num_nurses):
            s = shifts[(j, day)]
            solver.Add(s.IndexOf(nurses_for_day) == j)

    # Create the decision builder.
    db = solver.Phase(shifts_flat, solver.CHOOSE_FIRST_UNBOUND,
                    solver.ASSIGN_MIN_VALUE)
    # Create the solution collector.
    solution = solver.Assignment()
    solution.Add(shifts_flat)
    collector = solver.AllSolutionCollector(solution)

    solver.Solve(db, [collector])
    print("Solutions found:", collector.SolutionCount())
    print("Time:", solver.WallTime(), "ms")
    print()


if __name__ == "__main__":
    main()

Как видите, единственные ограничения, которые я сохранил, - это отношения между сменами и медсестрами. При num_nurses = 3, num_shifts = 3 и num_days = 1 решатель может найти 6 решений. Однако, если я изменю num_shifts на 2, решатель вернет 0 решений. Разве здесь не должно быть 3 решений (назначьте одну медсестру, оставьте два других неназначенными)?


person Pravesh Jain    schedule 14.11.2018    source источник


Ответы (2)


Как оказалось, это ограничение в том, как сейчас написано employee_scheduling. Сейчас его переписывают, и это должно быть сделано через пару недель.

https://github.com/google/or-tools/issues/932

person Pravesh Jain    schedule 14.11.2018

Я рекомендую эту версию расписания смен, которая реализует несколько различных ограничений:

https://github.com/google/or-tools/blob/master/examples/python/shift_scheduling_sat.py

person Laurent Perron    schedule 19.11.2018