age-gender-app / README.md
vola2004's picture
Upload 5 files
b77b86b verified
|
Raw
History Blame Contribute Delete
1.38 kB
metadata
title: 年代・性別推定システム
emoji: 🧠
colorFrom: blue
colorTo: purple
sdk: docker
pinned: false
license: mit
app_port: 5000

年代・性別推定システム

日本語テキストから年代と性別を推定するAIシステムです。BERTベースのモデルを使用して、入力されたテキストの特徴から年代(10代〜60代)と性別(男性・女性)を確率で予測します。

機能

  • 年代推定: 10代、20代、30代、40代、50代、60代の6つの年代を確率で予測
  • 性別推定: 男性・女性を確率で予測
  • リアルタイム予測: Webアプリケーションでリアルタイムに予測結果を表示

技術仕様

  • ベースモデル: cl-tohoku/bert-large-japanese
  • フレームワーク: PyTorch, Transformers
  • Webアプリ: Flask
  • デプロイ: Hugging Face Spaces

使用方法

  1. テキストボックスに日本語のテキストを入力
  2. 「推測」ボタンをクリック
  3. 年代と性別の確率が表示されます

モデル詳細

  • 年代モデル: 各年代を独立した二値分類器として学習
  • 性別モデル: 2クラス分類(男性・女性)
  • 学習データ: 日本語テキストデータセット
  • 精度: 年代推定 約79%、性別推定 約70%

ライセンス

MIT License