Efficientnet ParkinsonsPred

🚀 高效網絡帕金森病預測模型 🤗

本倉庫包含用於預測帕金森病的Hugging Face高效網絡（EfficientNet）模型，該模型通過分析患者的繪圖進行預測，準確率約為83%。本模型基於高效網絡和PyTorch構建。

🚀 快速開始

本模型使用高效網絡架構，通過分析患者的繪圖來預測其患帕金森病的可能性。如果你願意，歡迎提交拉取請求（Pull Request）參與貢獻。

✨ 主要特性

• 基於高效網絡架構，對帕金森病進行預測。
• 利用患者繪圖數據，預測準確率約達83%。

📦 安裝指南

文檔未提及安裝步驟，故跳過此章節。

💻 使用示例

基礎用法

import torch
from transformers import AutoModel
from torch import nn
from PIL import Image
import numpy as np

# Set the device
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

# Load the trained model
model = AutoModel.from_pretrained('/content/final')

# Move the model to the device
model = model.to(device)

# Load and resize new image(s)
image_size = (224, 224)
new_image = Image.open('/content/health.png').convert('RGB').resize(image_size)
new_image = np.array(new_image)
new_image = torch.from_numpy(new_image).transpose(0, 2).float().unsqueeze(0)

# Move the data to the device
new_image = new_image.to(device)

# Make predictions using the trained model
with torch.no_grad():
    predictions = model(new_image)
    logits = predictions.last_hidden_state
    logits = logits.view(logits.shape[0], -1)
    num_classes=2
    feature_reducer = nn.Linear(logits.shape[1], num_classes)

    logits = logits.to(device)
    feature_reducer = feature_reducer.to(device)

    logits = feature_reducer(logits)
    predicted_class = torch.argmax(logits, dim=1).item()
    confidence = torch.softmax(logits, dim=1)[0][predicted_class].item()
    if(predicted_class == 0):
        print(f'Predicted class: Parkinson\'s with confidence {confidence:.2f}')
    else:
        print(f'Predicted class: Healthy with confidence {confidence:.2f}')

📚 詳細文檔

數據集

訓練此模型所使用的數據集由 Kaggle 提供。

📄 許可證

本項目採用MIT許可證。