Python - Как нарисовать s-кривую на основе переменных x и y - PullRequest
0 голосов
/ 09 февраля 2020

Ссылка на набор данных: - https://en.wikipedia.org/wiki/Logistic_regression Привязка набора данных - введите описание изображения здесь

Я хочу нарисовать s-кривую для набора данных, который указан в этом ссылка на сайт. Ниже приведен код, который я пробовал.

Мой код: -

import pandas as pd
import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt
from sklearn.linear_model import LogisticRegression
from sklearn.metrics import r2_score
import statsmodels.api as sm

#df=pd.read_csv("Advertising_lin_reg.csv")
candidate = {'Hours':[0.50,0.75,1.00,1.25,1.50,1.75,1.75,2.00,2.25,2.50,2.75,3.00,3.25,3.50,4.00,4.25,4.50,4.75,5.00,5.50],
             'Result':[0,0,0,0,0,0,1,0,1,0,1,0,1,0,1,1,1,1,1,1]}

df=pd.DataFrame(candidate,columns=['Hours','Result'])
df.head()
x = df['Hours'].values.reshape(-1,1)
y = df['Result'].values.reshape(-1,1)
f = 1 / (1 + np.exp(y))
plt.plot(x, f)
plt.xlabel('x')
plt.ylabel('f(x)')
plt.show() 
...