文本分类模型-北京尧图网络

# !/usr/bin/env python3 import argparse import asyncio import json import os import random import time from datetime import datetime from pathlib import Path import httpx from openai import AsyncOpenAI # 配置 # LLM_BASE_URL os.getenv(LLM_BASE_URL) # LLM_MODEL os.getenv(LLM_MODEL) LLM_API_KEY os.getenv(LLM_API_KEY) or empty LLM_TIMEOUT float(os.getenv(LLM_TIMEOUT, 30)) # 对象种类 CATEGORIES [徽章, 戒指, 动物, 人物, 其他] CATEGORIES_STR 、.join(CATEGORIES) SYSTEM_PROMPT f你是一个文本分类专家。根据给定的文生图提示词判断主要对象的种类。可选种类{CATEGORIES_STR} 严格只输出一个词不要解释。 async def classify(text: str) - str: 分类提示词 if not text.strip(): return client AsyncOpenAI(api_keyLLM_API_KEY, base_urlLLM_BASE_URL, http_clienthttpx.AsyncClient(timeoutLLM_TIMEOUT), ) try: resp await client.chat.completions.create(modelLLM_MODEL, messages[{role: system, content: SYSTEM_PROMPT}, {role: user, content: text.strip()}, ], temperature0.1, max_tokens20, ) result resp.choices[0].message.content.strip() # 提取第一个词 for cat in CATEGORIES: if cat in result: return cat return result finally: await client.close() async def main(): parser argparse.ArgumentParser() parser.add_argument(--output, typePath, defaultNone) parser.add_argument(--seed, typeint, defaultNone) args parser.parse_args() test_prompts [ # 人物 A single 3D designer art toy figure, presented in a strict horizontal triptych layout with left side view, middle front view, and right rear view. Strictly limited to a palette of exactly 4 pure colors: 1. Pure white for the head, gloves, and boots; 2. Dark ultramarine blue for the torso, arms, and legs; 3. Crimson red for the flowing cape; 4. Silver-grey for the facial mask details and chest logo., # 徽章 A beautifully crafted enamel pin badge, featuring a detailed dragon design in gold and red colors, with subtle metallic sheen, professional product photography, white background, high resolution, studio lighting., # 戒指 A stunning platinum engagement ring with a 2-carat diamond center stone, surrounded by smaller pavé-set diamonds, intricate filigree details on the band, elegant jewelry photography, soft natural lighting, macro shot, 8K., # 动物 A majestic golden eagle soaring through the sky with wings fully extended, dramatic sunset background, sharp focus on feathers, hyper-realistic wildlife photography, National Geographic style, 8K, cinematic composition., # 其他 A futuristic skyscraper with a sleek glass facade, reflecting the surrounding cityscape, dramatic night lighting, long exposure photography, architectural marvel, modern minimalist design, high resolution, 8K., # 其他 A delicious gourmet burger on a wooden table, perfectly chargrilled beef patty, melted cheddar cheese, crispy bacon, fresh lettuce and tomato, steam rising, warm ambient lighting, food photography, shallow depth of field, 8K, mouth-watering., ] results [] # 逐个测试 for i, prompt in enumerate(test_prompts, 1): total_start time.time() predicted await classify(prompt) results.append({predicted: predicted}) print(f{i} {predicted} {time.time() - total_start:.2f}) if __name__ __main__: asyncio.run(main())

文本分类模型

相关推荐

NLP 模型评测与多任务性能对比：发布前检查失败路径与回滚

2025届必备的AI辅助论文平台横评

喜马拉雅FM下载器：跨平台音频保存解决方案

更多资讯

本地优先开源AI角色扮演工具Utopia：从Ollama到角色卡的完整实践

Java中的length：数组属性、String方法与集合size的底层原理

Ego-Planner实战：解决无人机撞树问题的避障指南

从零构建本地AI角色扮演工具：Ollama+FastAPI+Vue实战

Jetson Orin NX串口读取GNSS模块：NMEA解析与RTK定位实战

ADS AEL参数化设计：磁性元件建模与自动化实现

今日更新

开源模型端侧落地实战：量化、推理加速与Agent上下文管理

AI Evals实战指南：从零搭建LLM应用评估体系与CI/CD集成

Java采购管理系统实战：从数据库设计到事务一致性

本周精选

如何彻底离线使用 Laya-CoreML：Hugging Face 下载、缓存机制与 local_files_only 详解

基于1Panel 部署MySQL MCP 服务：TaoToken 统一 Key 接入与 config.toml 配置实战

崇州园区营销网站建设避坑指南：5大注意事项保备案安全

本月热门

基于YOLO的焊缝缺陷检测：从数据标注到工业部署全流程实践

分布式服务的决策记录

创业产品的首版范围

读完文章，想聊聊您的网站？