Назад к вопросам
JuniorТеорияЧасто
Что выведет эта программа?
import numpy as np
from sklearn.tree import DecisionTreeRegressor
X = np.array([[1], [2], [3], [4], [5]])
y = np.array([10, 20, 30, 40, 50])
model = DecisionTreeRegressor(random_state=0).fit(X, y)
print(model.predict(np.array([[100]]))[0])50.0
- Ждать
1000.0, будто дерево продолжит прямуюy = 10x:x = 100проходит все пороги так же, какx = 5, и попадает в крайний правый лист с константой50.0. - Ждать
30.0, среднее всех целей: дерево возвращает константу листа, в который попал объект, а не общее среднее, и в этом листе толькоx = 5.
- Что предскажет
LinearRegressionдля того же входа и почему это отличается от дерева? - Как всё же применить дерево для прогноза величины, которая растёт со временем?