batuta/oracle/knowledge_graph/
domain_mappings.rs1use super::super::types::ProblemDomain;
4use super::types::KnowledgeGraph;
5
6impl KnowledgeGraph {
7 pub(crate) fn initialize_domain_mappings(&mut self) {
9 use ProblemDomain::{
10 CMigration, DataPipeline, DeepLearning, DistributedCompute, GraphAnalytics, Inference,
11 LinearAlgebra, MediaProduction, ModelServing, Profiling, PythonMigration,
12 ShellMigration, SpeechRecognition, SupervisedLearning, Testing, UnsupervisedLearning,
13 Validation, VectorSearch,
14 };
15
16 self.domain_capabilities.insert(
17 SupervisedLearning,
18 vec![
19 "linear_regression".into(),
20 "logistic_regression".into(),
21 "decision_tree".into(),
22 "random_forest".into(),
23 "gbm".into(),
24 "naive_bayes".into(),
25 "knn".into(),
26 "svm".into(),
27 ],
28 );
29
30 self.domain_capabilities.insert(
31 UnsupervisedLearning,
32 vec!["kmeans".into(), "pca".into(), "dbscan".into(), "hierarchical".into()],
33 );
34
35 self.domain_capabilities.insert(
36 DeepLearning,
37 vec!["autograd".into(), "lora".into(), "qlora".into(), "quantization".into()],
38 );
39
40 self.domain_capabilities.insert(
41 Inference,
42 vec!["model_serving".into(), "batching".into(), "moe_routing".into()],
43 );
44
45 self.domain_capabilities.insert(
46 SpeechRecognition,
47 vec![
48 "speech_recognition".into(),
49 "streaming_transcription".into(),
50 "multilingual".into(),
51 "whisper_quantization".into(),
52 ],
53 );
54
55 self.domain_capabilities.insert(
56 LinearAlgebra,
57 vec!["vector_ops".into(), "matrix_ops".into(), "simd".into(), "gpu".into()],
58 );
59
60 self.domain_capabilities.insert(
61 VectorSearch,
62 vec!["vector_store".into(), "similarity_search".into(), "knn_search".into()],
63 );
64
65 self.domain_capabilities.insert(
66 GraphAnalytics,
67 vec!["pathfinding".into(), "centrality".into(), "community_detection".into()],
68 );
69
70 self.domain_capabilities.insert(
71 PythonMigration,
72 vec!["type_inference".into(), "sklearn_to_aprender".into(), "numpy_to_trueno".into()],
73 );
74
75 self.domain_capabilities
76 .insert(CMigration, vec!["ownership_inference".into(), "unsafe_elimination".into()]);
77
78 self.domain_capabilities
79 .insert(ShellMigration, vec!["script_conversion".into(), "cli_generation".into()]);
80
81 self.domain_capabilities.insert(
82 DistributedCompute,
83 vec![
84 "work_stealing".into(),
85 "cpu_executor".into(),
86 "gpu_executor".into(),
87 "remote_executor".into(),
88 ],
89 );
90
91 self.domain_capabilities.insert(
92 DataPipeline,
93 vec!["csv".into(), "parquet".into(), "json".into(), "streaming".into()],
94 );
95
96 self.domain_capabilities.insert(
97 ModelServing,
98 vec!["model_serving".into(), "lambda".into(), "container".into(), "edge".into()],
99 );
100
101 self.domain_capabilities.insert(
102 Testing,
103 vec![
104 "coverage_check".into(),
105 "mutation_testing".into(),
106 "tdg_scoring".into(),
107 "parity_checking".into(),
108 "oracle_generation".into(),
109 "falsification_testing".into(),
110 "quantization_drift".into(),
111 "roundtrip_validation".into(),
112 ],
113 );
114
115 self.domain_capabilities.insert(
116 Profiling,
117 vec!["syscall_trace".into(), "flamegraph".into(), "golden_trace_comparison".into()],
118 );
119
120 self.domain_capabilities.insert(
121 Validation,
122 vec![
123 "privacy_audit".into(),
124 "quality_gates".into(),
125 "complexity_analysis".into(),
126 "contract_parsing".into(),
127 "scaffold_generation".into(),
128 "kani_verification".into(),
129 "probar_generation".into(),
130 "traceability_audit".into(),
131 "binding_registry".into(),
132 ],
133 );
134
135 self.domain_capabilities.insert(
136 MediaProduction,
137 vec![
138 "video_rendering".into(),
139 "mlt_xml".into(),
140 "ffmpeg_encode".into(),
141 "transition_blend".into(),
142 "course_production".into(),
143 "audio_processing".into(),
144 "subtitle_burn_in".into(),
145 "media_concat".into(),
146 "transcription_qa".into(),
147 "vocabulary_enrichment".into(),
148 "content_generation".into(),
149 "svg_asset_pipeline".into(),
150 "course_quality_scoring".into(),
151 "course_publishing".into(),
152 ],
153 );
154 }
155}