このプロジェクトは、伝統的な機械学習(ML)モデルと現代的な大規模言語モデル(LLM)を一つの FastAPI アプリケーションに統合したオープンソースプロジェクトです。
このアプリケーションは、以下の2つの主要な機能を提供します:
- Iris 分類 (ML): Scikit-learn を使用して学習されたロジスティック回帰モデルを用い、花のガクと花弁のサイズから種類を予測します。
- 感情分析 (LLM): Hugging Face Transformers を使用し、入力されたテキストの感情(ポジティブ/ネガティブ)を分析します。
- Framework: FastAPI
- Machine Learning: Scikit-learn, Joblib, NumPy
- NLP/LLM: Transformers (DistilBERT), PyTorch
- Testing: Pytest, HTTPX
pip install -r requirements.txt既製の model.joblib が含まれていますが、以下のコマンドで再学習が可能です:
python train_ml_model.pyuvicorn main:app --reloadサーバーは http://localhost:8000 で起動します。
- URL:
/predict - Method:
POST - Payload:
{"features": [5.1, 3.5, 1.4, 0.2]}
- URL:
/sentiment - Method:
POST - Payload:
{"text": "FastAPI is amazing!"}
pytestこのプロジェクトはオープンソースです。コントリビューション、Issueの報告、フォークを歓迎します。
*Created by Jules & hombredennis66