機械学習コードの微編集

「python コードの編集」 Optunaを使ったコードがあります。 今は、2分類のコードですが、編集して3分類以上も出来るようにして頂きたいです。 また、分類する数を後で、編集できるように、#で記載していただきたいです。 コードはご提案後に送付します。　　 出来れば、5000円でお願い致します。