25ce3590ee
🔧 주요 개선사항: - EXCLUDE 분류기 추가 (WELD GAP 등 제외 대상 처리) - FITTING 분류기 키워드 확장 (ELL, RED 추가) - PIPE 재질 중복 문제 해결 (material_grade 파싱 개선) - NIPPLE 특별 처리 추가 (스케줄 + 길이 정보 포함) - OLET 타입 중복 표시 제거 📊 분류 정확도: - UNKNOWN: 0개 (100% 분류 성공) - EXCLUDE: 1,014개 (제외 대상) - 실제 자재: 1,823개 정확 분류 🎯 해결된 문제: - PIPE 재질 'ASTM A106 ASTM A106' → 'ASTM A106 GR B' - WELD GAP 오분류 → EXCLUDE 카테고리 - FITTING 키워드 인식 실패 → ELL, RED 키워드 추가 - 프론트엔드 중복 표시 제거
89 lines
3.0 KiB
Python
89 lines
3.0 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
main_nom, red_nom 기능 테스트 스크립트
|
|
"""
|
|
|
|
import sys
|
|
import os
|
|
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
from app.services.fitting_classifier import classify_fitting
|
|
from app.services.flange_classifier import classify_flange
|
|
|
|
def test_main_red_nom():
|
|
"""main_nom과 red_nom 분류 테스트"""
|
|
|
|
print("🔧 main_nom/red_nom 분류 테스트 시작!")
|
|
print("=" * 60)
|
|
|
|
test_cases = [
|
|
{
|
|
"name": "일반 TEE (동일 사이즈)",
|
|
"description": "TEE, SCH 40, ASTM A234 GR WPB",
|
|
"main_nom": "4\"",
|
|
"red_nom": None,
|
|
"expected": "EQUAL TEE"
|
|
},
|
|
{
|
|
"name": "리듀싱 TEE (다른 사이즈)",
|
|
"description": "TEE RED, SCH 40 x SCH 40, ASTM A234 GR WPB",
|
|
"main_nom": "4\"",
|
|
"red_nom": "2\"",
|
|
"expected": "REDUCING TEE"
|
|
},
|
|
{
|
|
"name": "동심 리듀서",
|
|
"description": "RED CONC, SCH 40 x SCH 40, ASTM A234 GR WPB",
|
|
"main_nom": "6\"",
|
|
"red_nom": "4\"",
|
|
"expected": "CONCENTRIC REDUCER"
|
|
},
|
|
{
|
|
"name": "리듀싱 플랜지",
|
|
"description": "FLG REDUCING, 300LB, ASTM A105",
|
|
"main_nom": "6\"",
|
|
"red_nom": "4\"",
|
|
"expected": "REDUCING FLANGE"
|
|
}
|
|
]
|
|
|
|
for i, test in enumerate(test_cases, 1):
|
|
print(f"\n{i}. {test['name']}")
|
|
print(f" 설명: {test['description']}")
|
|
print(f" MAIN_NOM: {test['main_nom']}")
|
|
print(f" RED_NOM: {test['red_nom']}")
|
|
|
|
# 피팅 분류 테스트
|
|
fitting_result = classify_fitting(
|
|
"",
|
|
test['description'],
|
|
test['main_nom'],
|
|
test['red_nom']
|
|
)
|
|
|
|
print(f" 🔧 FITTING 분류 결과:")
|
|
print(f" 카테고리: {fitting_result.get('category')}")
|
|
print(f" 타입: {fitting_result.get('fitting_type', {}).get('type')}")
|
|
print(f" 서브타입: {fitting_result.get('fitting_type', {}).get('subtype')}")
|
|
print(f" 신뢰도: {fitting_result.get('overall_confidence', 0):.2f}")
|
|
|
|
# 사이즈 정보 확인
|
|
size_info = fitting_result.get('size_info', {})
|
|
print(f" 메인 사이즈: {size_info.get('main_size')}")
|
|
print(f" 축소 사이즈: {size_info.get('reduced_size')}")
|
|
print(f" 사이즈 설명: {size_info.get('size_description')}")
|
|
|
|
# RED_NOM이 있는 경우 REDUCING 분류 확인
|
|
if test['red_nom']:
|
|
fitting_type = fitting_result.get('fitting_type', {})
|
|
if 'REDUCING' in fitting_type.get('subtype', '').upper():
|
|
print(f" ✅ REDUCING 타입 정상 인식!")
|
|
else:
|
|
print(f" ❌ REDUCING 타입 인식 실패")
|
|
|
|
print("-" * 50)
|
|
|
|
print("\n🎯 테스트 완료!")
|
|
|
|
if __name__ == "__main__":
|
|
test_main_red_nom() |