طراحی کد ماشین لرنینگی که برای تشخیص اشخاص بیمار

با سلام. یک دیتاییه مربوط به ocd, قراره روی یک کد ماشین لرنینگی که برای تشخیص اشخاص بیمار و سالم نوشته شده، پیاده بشه. ولی چون همه ی اشخاص ما بیمار هستن، پری پراسسینگ کد باید تغییر کنه که بجای تشخیص بیمار و سالم، اشخاص بیمار رو به سه شدت ضعیف و متوسط و قوی classify کنه. و سپس با cnn مشابه کدی که از قبل هست اجرا بشه. https://paperswithcode.com/paper/medformer-a-multi-granularity-patching کد در لینک بالا ذکر شده