ترین کد پایتون در پزشکی

مقاله ای را بررسی میکنم که در زمینه تقسیم بندی تصاویر پزشکی بوده و هدف تشخیص وجود تومور در مغز میباشد. تصاویر موجود بوده و mri هستند. مقاله مورد نظر در گیتهاب کد دارد. کد دارای باگ بوده و لازم است رفع شود. میخواهم مدل را train کنید و کد بدون باگ + نتایج حاصل از ترین را با من شیر کنید. کد کامل بوده و توضیحات در رابطه با ترین کردن آن در گیتهاب داده شده است. تنها لازم است باگ ها رفع شوند. مرحله ی preparation of data انجام شده و دیتای کامل رو باهاتون شیر میکنم. مرحله ی ترین لازم به اجرا توسط شماست. برای سورس کد پیام بدین