You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
24 lines
688 B
24 lines
688 B
from sklearn.linear_model import LinearRegression
|
|
import pandas as pd
|
|
import numpy as np
|
|
|
|
def generateModele(dataJson:dict[str,str]):
|
|
# -- Préparation des données
|
|
arrayBpm = []
|
|
arrayStartTime = []
|
|
arrayTimeOfActivity = []
|
|
|
|
for data in dataJson["Data"]:
|
|
arrayBpm.append(data["BpmAvg"])
|
|
arrayTimeOfActivity.append(data["TimeOfActivity"])
|
|
|
|
arrayStartTime.append(data["StartTime"])
|
|
# -- DataFrame
|
|
data = pd.DataFrame({
|
|
"Bpm": arrayBpm,
|
|
"TimeOfActivity": arrayTimeOfActivity
|
|
})
|
|
# -- Régression linéaire
|
|
model = LinearRegression()
|
|
model.fit(np.array(arrayStartTime).reshape(-1,1),data)
|
|
return model |