Как использовать sklearn python для получения прогноза

У меня есть таблица, и я хочу передать функции = "train_1, train_2, train_3, train_4" и target_result = result_cor.

Я хочу знать, когда значения = "1 или 2" в моем предсказание:

следить за моими данными

следовать мой code:

но результат всегда 100,0 % или 0,0. Мне нужно знать, сколько раз мой result_cor появляется в model_predict моей обученной модели

пожалуйста, помогите

from enum import auto
from sklearn.svm import LinearSVC
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
from sklearn.tree import DecisionTreeClassifier
from sklearn.metrics import classification_report 
from sklearn import svm
from sklearn.linear_model import LogisticRegression
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import math
import seaborn as sns

sheet_id = '1CfnVwuqysTYNPKLVhgjJ44Af8VDcdN1l'
dados = pd.read_excel(f'https://docs.google.com/spreadsheets/export?id={sheet_id}&format=xlsx')
dados.head()

# Dados para realizar o aprendizado e ver quanto vai prever corretamente
x = dados[['train_1','train_2','train_3','train_4']]

#  Gabarito ou resultados ja corretos
y = dados[['result_cor']]

#  Efetuo a separação dos treinos de x e y e testes de x e y
treino_x, teste_x, treino_y, teste_y = train_test_split(x,y,test_size=0.33)

# Tipo de modelo
modelo = DecisionTreeClassifier()

#  Efetuo o treinamento
modelo.fit(x,np.ravel(y,order="c"))

# Predicion new valor 
model_predict = [0,1,0,1]
treino_x[:1] = model_predict
model_predict = treino_x[:1]

result_cor = [1]
treino_y[:1] = result_cor 
result_cor = treino_y[:1]

# predicion the new model
previsoes = modelo.predict(model_predict)

#  Check acuracy 
accuracy = accuracy_score(result_cor,previsoes) * 100

print(f'A acuracia é: {round(accuracy,2)}')

Анжела
Вопрос задан11 июня 2024 г.

1 Ответ

Ваш ответ

Загрузить файл.