機械学習で回帰直線がうまく書けない

Question

機械学習初心者です。
以下のプログラムを書いたのですが、テスト結果の回帰直線が訓練用のデータ分布に沿わず、垂直になってしまいます。何がおかしいのかヒントをいただけますか？

from sklearn import datasets
from sklearn import linear_model
import numpy as np
import matplotlib.pyplot as plt
from pandas import DataFrame

boston = datasets.load_boston()
boston_df = DataFrame(boston.data)
boston_df.columns = boston.feature_names

rooms_train = DataFrame(boston_df["RM"])
y_train = boston.target
model = linear_model.LinearRegression()
model.fit(rooms_train, y_train)

rooms_test = DataFrame(np.arange(rooms_train.min(),rooms_train.max(),0.1))
prices_test = model.predict(rooms_test)

plt.scatter(rooms_train,y_train,c="b",alpha=0.5)

plt.plot(rooms_test,prices_test,c="r")

plt.title("Boston House Prices dataset")
plt.xlabel("rooms")
plt.ylabel("price $1000's")

Answer 1

pandas.DataFrameをnumpy.ndarrayに変換するとうまくいきました。

plt.scatter(rooms_train.values, y_train, c="b", alpha=0.5)
plt.plot(rooms_test.values, prices_test, c="r")

.valuesでndarrayになります。

Comments

[fuzzball]

(deleted)

[fuzzball]

ライブラリ側の不具合の臭いがするので環境を書いておきます。
macOS 10.12.6
Python 3.6.3
matplotlib 2.1.0
numpy 1.13.3
pandas 0.20.3

Answer 2

私の環境では、上記のコードのコピペで問題なく動きました。（win-32 python3.6）