error rate about ciw HOT 4 CLOSED

class ErroredService(ciw.dists.Distribution):
    def __init__(self, rate, error_prob):
        self.rate = rate
        self.error_prob = error_prob
    
    def sample(self, t, ind):
        full_service_time = random.expovariate(self.rate)
        rnd = random.random()
        if rnd < self.error_prob:
            return full_service_time * 0.5
        return full_service_time

>>> N1 = ciw.create_network(
...     arrival_distributions=[ciw.dists.Exponential(rate=5)],
...     service_distributions=[ErroredService(rate=6, error_prob=0.1)],
...     number_of_servers=[1]
... )
>>> ciw.seed(0)
>>> Q = ciw.Simulation(N1)
>>> Q.simulate_until_max_time(500)
>>> recs = Q.get_all_records()
>>> service_times = [r.service_time for r in recs]
>>> sum(service_times) / len(service_times)
0.16053608937056327

>>> N2 = ciw.create_network(
...     arrival_distributions=[ciw.dists.Exponential(rate=5)],
...     service_distributions=[ErroredService(rate=6, error_prob=0.9)],
...     number_of_servers=[1]
... )
>>> ciw.seed(0)
>>> Q = ciw.Simulation(N2)
>>> Q.simulate_until_max_time(500)
>>> recs = Q.get_all_records()
>>> service_times = [r.service_time for r in recs]
>>> sum(service_times) / len(service_times)
0.09299399922270359

class ErroredService(ciw.dists.Distribution):
    def __init__(self, rate, error_prob):
        self.rate = rate
        self.error_prob = error_prob
    
    def sample(self, t, ind):
        full_service_time = random.expovariate(self.rate)
        rnd = random.random()
        if rnd < self.error_prob:
            ind.errored = True
            return full_service_time * 0.5
        ind.errored = False
        return full_service_time

from collections import namedtuple
ErroredDataRecord = namedtuple('Record', [
    'id_number',
    'customer_class',
    'original_customer_class',
    'node',
    'arrival_date',
    'waiting_time',
    'service_start_date',
    'service_time',
    'errored_service',
    'service_end_date',
    'time_blocked',
    'exit_date',
    'destination',
    'queue_size_at_arrival',
    'queue_size_at_departure',
    'server_id',
    'record_type'
    ])

import math
class ErroredNode(ciw.Node):
    def write_individual_record(self, individual):
        if math.isinf(self.c):
            server_id = False
        else:
            server_id = individual.server.id_number
        
        record = ErroredDataRecord(
            individual.id_number,
            individual.previous_class,
            individual.original_class,
            self.id_number,
            individual.arrival_date,
            individual.service_start_date - individual.arrival_date,
            individual.service_start_date,
            individual.service_end_date - individual.service_start_date,
            individual.errored,
            individual.service_end_date,
            individual.exit_date - individual.service_end_date,
            individual.exit_date,
            individual.destination,
            individual.queue_size_at_arrival,
            individual.queue_size_at_departure,
            server_id,
            'service')
        individual.data_records.append(record)

>>> N1 = ciw.create_network(
...     arrival_distributions=[ciw.dists.Exponential(rate=5)],
...     service_distributions=[ErroredService(rate=6, error_prob=0.1)],
...     number_of_servers=[1]
... )
>>> ciw.seed(0)
>>> Q = ciw.Simulation(N1, node_class=ErroredNode)
>>> Q.simulate_until_max_time(500)
>>> recs = Q.get_all_records()
>>> errored_serviced = [r.errored_service for r in recs]
>>> sum(errored_serviced) / len(errored_serviced)
0.10408326661329063

>>> N2 = ciw.create_network(
...     arrival_distributions=[ciw.dists.Exponential(rate=5)],
...     service_distributions=[ErroredService(rate=6, error_prob=0.9)],
...     number_of_servers=[1]
... )
>>> ciw.seed(0)
>>> Q = ciw.Simulation(N2, node_class=ErroredNode)
>>> Q.simulate_until_max_time(500)
>>> recs = Q.get_all_records()
>>> errored_serviced = [r.errored_service for r in recs]
>>> sum(errored_serviced) / len(errored_serviced)
0.9032128514056225