37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
from dataclasses import dataclass, field
|
|
from datetime import datetime
|
|
|
|
|
|
@dataclass
|
|
class TCalcDatasetConfig:
|
|
start_date: 'datetime' = datetime(year=2023, month=1, day=1)
|
|
date_format: str = "%Y-%m-%d"
|
|
datetime_format: str = "%Y-%m-%d %H:%M:%S"
|
|
|
|
input_columns: list = field(default_factory=lambda: [
|
|
"created_at", "pick_at", "from_lat", "from_lon",
|
|
"to_lat", "to_lon", "weight", "volume"
|
|
])
|
|
|
|
preprocessing: list[dict] = field(default_factory=lambda: [
|
|
{"name": "from_lat", "bias": 40, "std": 4},
|
|
{"name": "from_lon", "bias": 40, "std": 4},
|
|
{"name": "to_lat", "bias": 40, "std": 4},
|
|
{"name": "to_lon", "bias": 40, "std": 4},
|
|
{"name": "distance", "bias": 700, "std": 950},
|
|
{"name": "weight", "bias": 12000, "std": 8000},
|
|
{"name": "volume", "bias": 55, "std": 31},
|
|
{"name": "time_passed", "bias": 6, "std": 3},
|
|
{"name": "car_type_id", "bias": 1, "std": 1},
|
|
{"name": "urgency", "bias": 0, "std": 1},
|
|
])
|
|
distributions: dict = field(default_factory=lambda: {
|
|
"dot": [1, 0, 0],
|
|
"trades": [0, 1, 0],
|
|
"tariff": [0, 0, 1]
|
|
})
|
|
use_distr: str = "dot"
|
|
coordinates_thr: dict = field(default_factory=lambda: {
|
|
"lat": [39.6, 72.89],
|
|
"lon": [19.32, 158.34]
|
|
}) |