Tensorflow And Pytorch

Applications in tensorflow and pytorch.

Tensorflow

1. transform “pandas.dataframe” to “tensorflow.dataset”

For training dataset, you should set the “label” to be the prediction label. In the “Titanic Survivors Prediction”, the label should be set to “Survived”.

1
# tfdf: tensorflow_decision_forests
2

3
train_ds = tfdf.keras.pd_dataframe_to_tf_dataset(pandans_train_dataframe, label="Survived")
4
test_ds = tfdf.keras.pd_dataframe_to_tf_dataset(pandans_test_dataframe)

PyTorch

1. flatten (扁平化) torch.flatten(tensor)