From 4d660f9b8ce9d76254e04626e15f078faef4fbc4 Mon Sep 17 00:00:00 2001 From: haotian <2421912570@qq.com> Date: Tue, 25 Feb 2025 10:25:14 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B5=8B=E8=AF=95--=E6=A8=A1=E5=9E=8B=E9=A2=84?= =?UTF-8?q?=E6=B5=8B=E6=8E=A5=E5=8F=A3=E6=B5=8B=E8=AF=95=E5=AE=8C=E6=88=90?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- api/__pycache__/model_api.cpython-39.pyc | Bin 3712 -> 3728 bytes api/model_api.py | 4 ++-- doc/安装文档.md | 29 ++++++++++++++++++++--- 3 files changed, 28 insertions(+), 5 deletions(-) diff --git a/api/__pycache__/model_api.cpython-39.pyc b/api/__pycache__/model_api.cpython-39.pyc index b21ca51bd80680dbb16f68cac8d8196a019f152b..bbc408fd29886de9a72a94c0792c5f315e88f10c 100644 GIT binary patch delta 1453 zcmaJ>%}*0S6yMq1?ryi+4+JemO85}80ihT_6GW^f#)}8=;6c+Y>{tr@a&|FFAT^q3 zjF6a&M`L25i5xKTZ2S+rdYO|k@#8Ph6N&HbHWm%g&HQHGn>W9C^FDe%{VW}|qft|W z->?3U#n~N{y2k8Mb%x$vrmiPU52Zrg z_7;S3$1Vq}uw5?JSjk(ch_Gk7O9k&5Zv~6)rqSVKnJg?Ns{7^)K8I?O((w@O0}n9-!%Xv#UK_* z5H`x6gw~UTpkYG@M-h$zWE3{cN3@>o7;26q>|jGN3xEfkgl0gPp?P6%WT zWPL+^0DD*j%b1bxZ(n_T`Sjt& zB5Ol^>s1Ym6Xw%(?IH~V7G}e!j0E-NpKj5$E@KI#^7-)fn=(6rirpf78a7VBa51@( zq7wLIAN6fjQ-5m_{>B(1=lEx1y>T8)c6jI~5)c2)CagsqgWoD-*G~Y1ViFB6AnfMH z6l`3E5hO#41(`?9KVX*x!X55Y51FF?y1& z5*fd3E#y~mj>jV>Nf*yYJlaaq?jf)stI*MfAVHXdjLf=V+V$yq*qDRStixuJ=Xu(C z-0%+_#*^I$*txBy4cc{2XUw2j^g5$XY2FJQ*q_Vql{0ur&?|oV`fWcA%0)gN?IY8C zHd=GA4p{^N{}6-%NoS)d`hPRnw+#DBsP&rbF|0B+fG~xCkrrAIB!IO-`@aHyuBX;Bu$gmJB delta 1465 zcmaJ>Noy2A6z=ZoWu`MT8O<))#ze>@W>HKc8jM*;Jcx)2UWB2Ot_%|=Gvn({5J`xr z;K2mMx=Jty!5kC>b5al9{0)8ZAh_JbA28s1)iF9DNf-51)mK&Ty?#qQZMokPcj9qd z!S7MuhiuK;xUH$@M2XdB49+;4)0z~nyV{7t4ICMcCTEhrALE;-;VaG!%0SU5s%*1M9n#I`n?01FjwDNu(4)QPe-5sTo| zRwHT>ySU1}u0l03={O|m%n^d41T_S;1kDHm%N2ZRrDYo}^_gNZFK>cLBz{{1`{+Df3Tm_5&p1m?F7M3bc}D)Q*F;jtdWDgMqzXOa zow3$%92w{!=p;CSkY>;&ZknwFCyD7MsMG+x$fdCF2(&|iz2c?m9VoEhs_#kfBdZTN z;wz2o6D4b4^)m9lGOsQx^O~zRDKx%duC_^xj@`g+y6ib~^>wzQE~~E5q`2m74KDeX zYvDSA>*%`n0QWE(?8O+0{Qhe5+s2chZ(jd+|1dCoN{@goIeGkX`OD;Niq0y!BCzFt zhVmtd;Gu99lU~it77KH^EL1v?^2qa?EDyaIfi9F6RE2WnJ%|&-Fit@Bz+Z(5FkEhF z&`)glX~I3OA42XFc8@}>Fdg&3)^rtHSudMW@hQ@e74kc>mb!rZdn|)N;&x-a5p5nE zCFUZ*et{mt4eDWV%tRRIG4U(f*%(ywyCn{}1zzwo1*{B{fbGE8Vh3`A55b ze1c9LZowoTp)~NQ3=~UOHELzkkU2al+N#dzjkqJe80W>asv&-uQal&Lu>7&y!?KY6 zPHR9C_I1D%Zrs9NndwuaH}-f{UZT2!V+2&_|7tv}?gNdn!=$iEz}?)e9~yADEJ_VH zOAw}nOd9!0Oe5)vv3N?I7PIl9OZf&>1oVEuFhLK2e3htpfvPUhOGVEI$|z_jphyEb mJTOBi0sTIZ`2W&SYOfbViH;gG8O8Lo@HxaT$7+#Jbp8PxC=C(- diff --git a/api/model_api.py b/api/model_api.py index 6e0b801..83f2bf3 100644 --- a/api/model_api.py +++ b/api/model_api.py @@ -14,7 +14,7 @@ class TrainRequest(BaseModel): algorithm: str task_type : str parameters: Dict - metrics: List[str] + # metrics: List[str] experiment_name: str class PredictRequest(BaseModel): @@ -60,7 +60,7 @@ async def train_model(request: TrainRequest): 'algorithm': request.algorithm, 'task_type': request.task_type, 'params': request.parameters, - 'metrics': request.metrics + # 'metrics': request.metrics }, experiment_name=request.experiment_name ) diff --git a/doc/安装文档.md b/doc/安装文档.md index 3cd1652..29a3099 100644 --- a/doc/安装文档.md +++ b/doc/安装文档.md @@ -61,19 +61,42 @@ GET http://10.0.0.202:8992/model/available/{model_name} GET http://10.0.0.202:8992/model/metrics ### 3.10 模型训练 -POST +POST http://10.0.0.202:8992/model/train + +{ + "train_path": "/home/admin-root/haotian/MLPlatform/dataset/dataset_processed/breast_cancer_20250224_170615/train_breast_cancer_20250224_170615.csv", + "val_path": "/home/admin-root/haotian/MLPlatform/dataset/dataset_processed/breast_cancer_20250224_170615/val_breast_cancer_20250224_170615.csv", + "algorithm": "XGBClassifier", + "task_type": "classification", + "parameters": { + "n_estimators": 100, + "learning_rate": 0.1, + "max_depth": 6, + "random_state": 42 + }, + "experiment_name": "test_post_1" +} ### 3.11 获取实验列表 GET http://10.0.0.202:8992/model/experiments ### 3.12 获取具体实验内容 -GET http://10.0.0.202://8992/model/experiment/{experiment_name} +GET http://10.0.0.202:8992/model/experiment/{experiment_name} ### 3.13 删除模型指定实验模型 DELETE ### 3.14 模型预测 -POST +POST http://10.0.0.202:8992/model/predict + { + "run_id": "33939ea6d8ce4d43a268f23f7361651e", + "data_path": "/home/admin-root/haotian/MLPlatform/dataset/dataset_processed/breast_cancer_20250219_145614/test_breast_cancer_20250219_145614.csv", + "output_path": "predictions/pred_breast_cancer_20250219_145614.csv", + "return_proba": true, + "metrics": [ + "accuracy", "f1", "precision", "recall" + ] + } ### 3.15 查看系统资源使用情况 GET http://10.0.0.202:8992/system/resources