オリジナルのデータセットを確認したい

Question

前提・実現したいこと

オリジナル画像のデータセットを作ろうとしたため、以下の方の記事を参考にしました。
ここでは画像ファイルのdogとcatを、自分用のgomaとsioに置き換えて、「データセット作成：カラー画像の場合」を行いました。

https://intellectual-curiosity.tokyo/2019/07/02/%E3%82%AA%E3%83%AA%E3%82%B8%E3%83%8A%E3%83%AB%E3%81%AE%E7%94%BB%E5%83%8F%E3%81%8B%E3%82%89%E3%83%87%E3%83%BC%E3%82%BF%E3%82%BB%E3%83%83%E3%83%88%E3%82%92%E4%BD%9C%E6%88%90%E3%81%99%E3%82%8B%E6%96%B9/

しかし、エラーが出てうまくいきません。
これと似たエラーを調べると's'に因数が必要なようですが、's'がどれを指すのかが分からず、因数をどのように扱えば良いか分かりません。
どうすれば解決できるでしょうか。

発生している問題・エラーメッセージ

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-8-ff5fb586616f> in <module>()
     36     plt.subplot(2, 2, i+1)
     37     plt.axis('off')
---> 38     plt.title(label = 'goma' if y_train[i] == 0 else 'sio')
     39     img_array = cv2.cvtColor(X_train[i], cv2.COLOR_BGR2RGB)
     40     plt.imshow(img_array)

TypeError: title() missing 1 required positional argument: 's'

該当のソースコード

import matplotlib.pyplot as plt
import os
import cv2
import random
import numpy as np

DATADIR = "C:/pan"
CATEGORIES = ["goma", "sio"]

IMG_SIZE = 50
training_data = []

def create_training_data():
    for class_num, category in enumerate(CATEGORIES):
        path = os.path.join(DATADIR, category)
        for image_name in os.listdir(path):
            try:
                img_array = cv2.imread(os.path.join(path, image_name),)  # 画像読み込み
                img_resize_array = cv2.resize(img_array, (IMG_SIZE, IMG_SIZE))  # 画像のリサイズ
                training_data.append([img_resize_array, class_num])  # 画像データ、ラベル情報を追加
            except Exception as e:
                pass
            
create_training_data()

random.shuffle(training_data)  # データをシャッフル
X_train = []  # 画像データ
y_train = []  # ラベル情報

# データセット作成
for feature, label in training_data:
    X_train.append(feature)
    y_train.append(label)
    
# numpy配列に変換
X_train = np.array(X_train)
y_train = np.array(y_train)

# データセットの確認
for i in range(0, 4):
    print("学習データのラベル：", y_train[i])
    plt.subplot(2, 2, i+1)
    plt.axis('off')
    plt.title(label = 'goma' if y_train[i] == 0 else 'sio')
    img_array = cv2.cvtColor(X_train[i], cv2.COLOR_BGR2RGB)
    plt.imshow(img_array)
    
plt.show()

補足情報（FW/ツールのバージョンなど）

使用環境
anaconda(python3)

Answer 1

matplotlibのバージョンが古くないですか？
2.x.x系統だと質問文の書き方ではエラーになると思います。
3.x.xだと問題ないと思いますが。

# version 2.x.xならこの書き方
plt.title('goma' if y_train == 0 else 'sio')

特に不都合がないならバージョン3に上げたほうが良いと思います。python3ですよね？

Comments

[eg88]

ご回答ありがとうございます！
ご指摘の通りmatplotlibをバージョンアップした後に %matplotlib inline のコードを加えた結果、うまくいきました。