Pasted

import pandas as pd
import numpy as np
from sklearn.preprocessing import LabelEncoder, OneHotEncoder
from sklearn.compose import ColumnTransformer
from keras.models import Sequential
from keras.layers import Dense, Dropout
 
#Busca
csv = pd.read_csv('dados.csv', sep=';')
 
#Remove colunas desnecessárias
csv = csv.drop(columns=['Número comentários', 'Compartilhamento'])
 
#Trata
le = LabelEncoder()
csv['Tipo'] = le.fit_transform(csv['Tipo']) #Foto = 0|Link = 1|Status = 2|Video = 3
dados = csv.values
 
#Separa em Atributos e Classificadores
atributos = dados[:,:5]
classificadores = dados[:,5]
 
#Ajusta atributos de classificação não binária
ct = ColumnTransformer([('binario', OneHotEncoder(), [0, 1])], remainder='passthrough')
atributos = ct.fit_transform(atributos).toarray()
 
#Modelo
modelo = Sequential()
modelo.add(Dense(units=10, activation='relu'))
modelo.add(Dense(units=1, activation='linear'))
modelo.compile(loss = 'mean_absolute_error', optimizer = 'adam',
                  metrics = ['mean_absolute_error'])
 
modelo.fit(atributos, classificadores, batch_size=50, epochs=1000)
 
#Predizendo
#Tipo, Mes, Dia da Semana, Hora, Paga
novo = np.array([
    [0, 1, 7, 21, 1]
])
novo = ct.transform(novo).toarray()
 
retorno = modelo.predict(novo)
print('Média: ', int(retorno[0]))