How to download ei_tensorflow.training

Hi!
I found that in NN classifier, there is a python packet named"ei_tensorflow.training", how to download it?
Thanks!

Hi @sherry,

This is an internal package we use for training the neural networks.
If you need to export the iPython notebook, you can replace some functions to bypass the ei_tensorflow package, see this topic: Keras(export) mode error

We are also working on improving our iPython notebook export so it doesn’t require any external dependencies.

Aurelien

1 Like

I got it !
Thank you so much!

2 Likes

Hi @aurel, all,
It would appear that with the recent addition of the “auto-balance dataset” feature, a dependency on ei_impulse package has been reintroduced in the Python notebook export feature:

# train the neural network
model.compile(loss='categorical_crossentropy', optimizer=opt, metrics=['accuracy'])
model.fit(train_dataset, epochs=30, validation_data=validation_dataset, verbose=2, callbacks=callbacks, class_weight=ei_tensorflow.training.get_class_weights(Y_train))

Hello @kartben2,

Our ML team is working on a fix.
Thanks for notifying us.

Regards,

Louis

1 Like