retirada do escopo if __name__ no arquivo som.py

This commit is contained in:
matheus_cascalho 2020-12-07 15:38:48 -03:00
parent 20a4b9dfef
commit b9995e780a

View File

@ -83,10 +83,3 @@ class SOMTransformation(Transformation):
self.net.nodes_graph(colnum=colnum) self.net.nodes_graph(colnum=colnum)
else: else:
self.net.diff_graph() self.net.diff_graph()
if __name__ == '__main__':
file = '/home/matheus_cascalho/Documentos/matheus_cascalho/MINDS/TimeSeries_Lab/SOM/Appliance Energy Prediction/energydata_complete.csv'
df = pd.read_csv(file, index_col=0)
som = SOMTransformation(grid_dimension=(20, 20))
new_df = som.apply(df.iloc[:50], endogen_variable='Appliances', epochs=100, leaning_rate=0.1)
print(new_df.head())