Расчет прогнозов вручную с использованием отдельных фреймов данных

Question

Я разработал модель и у меня есть фрейм данных, содержащий используемые функции и их коэффициент:

feature    coefficient
intercept  2.5
a          0.5
b          0.8
c          1.7

У меня также есть фрейм данных, содержащий совершенно новые данные:

a  b  c
1  0  2
0  1  3

Я хочу создать прогноз для каждого наблюдения, используя формулу:

intercept + a_coef*a_value + b_coef*b_value + c_coef*c_value

Но я не уверен, как таким образом управлять обоими фреймами данных.

Answer 1

Вы можете назначить intercept в новом фрейме данных, а затем сделать dot

s=df.set_index('feature')['coefficient']
newdf.assign(intercept=1).dot(s)
Out[249]: 
0    6.4
1    8.4
dtype: float64