File size: 1,617 Bytes
f158aab |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 |
"""
Data Package
=============
Data handling, schema definitions, and dataset generation for
the Traffic Accident Reconstruction System.
"""
from .schema import (
# Enums
VehicleType,
Direction,
VehicleAction,
WeatherCondition,
RoadCondition,
AccidentType,
Severity,
RoadType,
# Data classes
Location,
Conditions,
VehicleData,
AccidentDetails,
ScenarioMetrics,
Scenario,
FaultAssessment,
TimelineEvent,
AccidentRecord,
AnalysisResult,
# Feature schema
FEATURE_SCHEMA,
# Validation functions
validate_speed,
validate_coordinates,
validate_path,
validate_probability
)
from .synthetic_dataset_generator import (
generate_dataset,
generate_single_accident,
generate_training_features,
save_dataset,
print_dataset_statistics
)
__all__ = [
# Enums
'VehicleType',
'Direction',
'VehicleAction',
'WeatherCondition',
'RoadCondition',
'AccidentType',
'Severity',
'RoadType',
# Data classes
'Location',
'Conditions',
'VehicleData',
'AccidentDetails',
'ScenarioMetrics',
'Scenario',
'FaultAssessment',
'TimelineEvent',
'AccidentRecord',
'AnalysisResult',
# Feature schema
'FEATURE_SCHEMA',
# Validation
'validate_speed',
'validate_coordinates',
'validate_path',
'validate_probability',
# Dataset generation
'generate_dataset',
'generate_single_accident',
'generate_training_features',
'save_dataset',
'print_dataset_statistics'
]
|