Source code for co2mpas.core.model.selector.models.engine_speed_model

# -*- coding: utf-8 -*-
#
# Copyright 2015-2019 European Commission (JRC);
# Licensed under the EUPL (the 'Licence');
# You may not use this work except in compliance with the Licence.
# You may obtain a copy of the Licence at: http://ec.europa.eu/idabc/eupl
"""
Functions and constants to define the engine_speed_model selector.
"""
import schedula as sh
from ._core import define_sub_model
from ...physical import dsp as _physical
from ...physical.clutch_tc import calculate_clutch_phases

#: Model name.
name = 'engine_speed_model'

#: Parameters that constitute the model.
models = [
    'final_drive_ratios', 'gear_box_ratios', 'idle_engine_speed_median',
    'idle_engine_speed_std', 'CVT', 'max_speed_velocity_ratio',
    'tyre_dynamic_rolling_coefficient'
]

#: Inputs required to run the model.
inputs = [
    'velocities', 'gears', 'times', 'on_engine', 'gear_box_type',
    'accelerations', 'tyre_code', 'after_treatment_speeds_delta',
    'motive_powers', 'accelerations', 'hybrid_modes', 'motor_p4_front_powers',
    'motor_p4_rear_powers', 'motor_p3_front_powers', 'motor_p3_rear_powers',
]

#: Relevant outputs of the model.
outputs = ['gear_box_speeds_in']

#: Targets to compare the outputs of the model.
targets = ['engine_speeds_out']

#: Extra inputs for the metrics.
metrics_inputs = [
    'times', 'velocities', 'gear_shifts', 'on_engine', 'stop_velocity',
    'hybrid_modes', 'after_treatment_warm_up_phases'
]


[docs]def metric_engine_speed_model( y_true, y_pred, times, velocities, gear_shifts, on_engine, stop_velocity, hybrid_modes, after_treatment_warm_up_phases=False): """ Metric for the `engine_speed_model`. :param y_true: Reference engine speed vector [RPM]. :type y_true: numpy.array :param y_pred: Predicted engine speed vector [RPM]. :type y_pred: numpy.array :param times: Time vector [s]. :type times: numpy.array :param velocities: Velocity vector [km/h]. :type velocities: numpy.array :param gear_shifts: When there is a gear shifting [-]. :type gear_shifts: numpy.array :param on_engine: If the engine is on [-]. :type on_engine: numpy.array :param stop_velocity: Maximum velocity to consider the vehicle stopped [km/h]. :type stop_velocity: float :param hybrid_modes: Hybrid mode status (0: EV, 1: Parallel, 2: Serial). :type hybrid_modes: numpy.array :param after_treatment_warm_up_phases: Phases when engine speed is affected by the after treatment warm up [-]. :type after_treatment_warm_up_phases: numpy.array :return: Error. :rtype: float """ from co2mpas.utils import mae b = ~calculate_clutch_phases(times, 1, 1, gear_shifts, 0, (-4.0, 4.0)) b &= (velocities > stop_velocity) & ~after_treatment_warm_up_phases b &= on_engine & (hybrid_modes == 1) return b.any() and float(mae(y_true[b], y_pred[b])) or .0
#: Metrics to compare outputs with targets. metrics = sh.map_list(targets, metric_engine_speed_model) #: Upper score limits to raise the warnings. up_limit = sh.map_list(targets, 40)
[docs]def select_models(keys, data): """ Select models from data. :param keys: Model keys. :type keys: list :param data: Cycle data. :type data: dict :return: Models. :rtype: dict """ mdl = sh.selector(keys, data, allow_miss=True) if 'tyre_dynamic_rolling_coefficient' in mdl: mdl.pop('r_dynamic', None) return mdl
#: Prediction model. # noinspection PyProtectedMember dsp = sh.Blueprint(_physical, inputs, outputs, models)._set_cls( define_sub_model )