latest
Get Started
Installation
Introduction by Example
Modular Design of Deep Tabular Models
Handling Advanced Stypes
Handling Heterogeneous Semantic Types
Handling Text Columns
Package Reference
torch_frame
torch_frame.data
torch_frame.datasets
torch_frame.nn
torch_frame.gbdt
torch_frame.config
torch_frame.transforms
torch_frame.utils
pytorch-frame
Index
Edit on GitHub
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
K
|
L
|
M
|
N
|
O
|
P
|
Q
|
R
|
S
|
T
|
V
|
X
|
Y
|
Z
A
ACCURACY (Metric attribute)
AdultCensusIncome (class in torch_frame.datasets)
AmazonFineFoodReviews (class in torch_frame.datasets)
B
BankMarketing (class in torch_frame.datasets)
BaseTransform (class in torch_frame.transforms)
BINARY_CLASSIFICATION (TaskType attribute)
C
cat() (in module torch_frame.utils)
(MultiEmbeddingTensor static method)
(MultiNestedTensor static method)
CatBoost (class in torch_frame.gbdt)
categorical (in module torch_frame.stype)
CatToNumTransform (class in torch_frame.transforms)
col_select() (Dataset method)
col_stats (Dataset property)
compute_metric() (GBDT method)
COUNT (StatType attribute)
CyclicEncoding (class in torch_frame.nn.encoding)
D
DataFrameBenchmark (class in torch_frame.datasets)
DataFrameTextBenchmark (class in torch_frame.datasets)
DataFrameToTensorFrameConverter (class in torch_frame.data)
DataLoader (class in torch_frame.data)
Dataset (class in torch_frame.data)
datasets_available() (DataFrameBenchmark class method)
(DataFrameTextBenchmark class method)
Decoder (class in torch_frame.nn.decoder)
device (TensorFrame property)
DiamondImages (class in torch_frame.datasets)
Dota2 (class in torch_frame.datasets)
download_url() (Dataset static method)
(in module torch_frame.data)
E
embedding (in module torch_frame.stype)
EmbeddingEncoder (class in torch_frame.nn.encoder)
encode_forward() (EmbeddingEncoder method)
(ExcelFormerEncoder method)
(LinearBucketEncoder method)
(LinearEmbeddingEncoder method)
(LinearEncoder method)
(LinearModelEncoder method)
(LinearPeriodicEncoder method)
(MultiCategoricalEmbeddingEncoder method)
(StackEncoder method)
(StypeEncoder method)
(TimestampEncoder method)
ExcelFormer (class in torch_frame.nn.models)
ExcelFormerConv (class in torch_frame.nn.conv)
ExcelFormerDecoder (class in torch_frame.nn.decoder)
ExcelFormerEncoder (class in torch_frame.nn.encoder)
F
FakeDataset (class in torch_frame.datasets)
feat_cols (Dataset property)
FeatureEncoder (class in torch_frame.nn.encoder)
fillna_col() (MultiEmbeddingTensor method)
(MultiNestedTensor method)
fit() (FittableBaseTransform method)
FittableBaseTransform (class in torch_frame.transforms)
ForestCoverType (class in torch_frame.datasets)
forward() (BaseTransform method)
(CyclicEncoding method)
(Decoder method)
(ExcelFormer method)
(ExcelFormerConv method)
(ExcelFormerDecoder method)
(FeatureEncoder method)
(FittableBaseTransform method)
(FTTransformer method)
(FTTransformerConvs method)
(MLP method)
(PositionalEncoding method)
(ResNet method)
(StypeEncoder method)
(StypeWiseFeatureEncoder method)
(TableConv method)
(TabNet method)
(TabTransformer method)
(TabTransformerConv method)
(Trompt method)
(TromptConv method)
(TromptDecoder method)
forward_embed() (ImageEmbedder method)
forward_retrieve() (ImageEmbedder method)
forward_stacked() (Trompt method)
from_tensor_list() (MultiEmbeddingTensor class method)
from_tensor_mat() (MultiNestedTensor class method)
FTTransformer (class in torch_frame.nn.models)
FTTransformerConvs (class in torch_frame.nn.conv)
G
GBDT (class in torch_frame.gbdt)
generate_random_split() (in module torch_frame.utils)
get_col_feat() (TensorFrame method)
get_split() (Dataset method)
H
HuggingFaceDatasetDict (class in torch_frame.datasets)
I
image_embedded (in module torch_frame.stype)
ImageEmbedder (class in torch_frame.config)
ImageEmbedderConfig (class in torch_frame.config)
index_select() (Dataset method)
infer_df_stype() (in module torch_frame.utils)
infer_series_stype() (in module torch_frame.utils)
is_empty (TensorFrame property)
is_fitted (FittableBaseTransform property)
(GBDT property)
is_materialized (Dataset property)
K
KDDCensusIncome (class in torch_frame.datasets)
L
LightGBM (class in torch_frame.gbdt)
LinearBucketEncoder (class in torch_frame.nn.encoder)
LinearEmbeddingEncoder (class in torch_frame.nn.encoder)
LinearEncoder (class in torch_frame.nn.encoder)
LinearModelEncoder (class in torch_frame.nn.encoder)
LinearPeriodicEncoder (class in torch_frame.nn.encoder)
load() (GBDT method)
(in module torch_frame.utils)
M
MAE (Metric attribute)
materialize() (DataFrameBenchmark method)
(DataFrameTextBenchmark method)
(Dataset method)
MEAN (NAStrategy attribute)
(StatType attribute)
Mercari (class in torch_frame.datasets)
Metric (class in torch_frame.typing)
MLP (class in torch_frame.nn.models)
ModelConfig (class in torch_frame.config)
module
torch_frame.stype
torch_frame.typing
MOST_FREQUENT (NAStrategy attribute)
MULTI_COUNT (StatType attribute)
multicategorical (in module torch_frame.stype)
MultiCategoricalEmbeddingEncoder (class in torch_frame.nn.encoder)
MULTICLASS_CLASSIFICATION (TaskType attribute)
MultiEmbeddingTensor (class in torch_frame.data)
MultimodalTextBenchmark (class in torch_frame.datasets)
MultiNestedTensor (class in torch_frame.data)
Mushroom (class in torch_frame.datasets)
MutualInformationSort (class in torch_frame.transforms)
N
na_forward() (StypeEncoder method)
NAStrategy (class in torch_frame.typing)
num_cols (TensorFrame property)
num_datasets_available() (DataFrameBenchmark class method)
(DataFrameTextBenchmark class method)
num_rows (Dataset property)
(TensorFrame property)
numerical (in module torch_frame.stype)
O
objective() (CatBoost method)
(LightGBM method)
(XGBoost method)
P
PokerHand (class in torch_frame.datasets)
PositionalEncoding (class in torch_frame.nn.encoding)
post_forward() (StypeEncoder method)
predict() (GBDT method)
Q
QUANTILES (StatType attribute)
R
REGRESSION (TaskType attribute)
reset_parameters() (Decoder method)
(EmbeddingEncoder method)
(ExcelFormerConv method)
(ExcelFormerDecoder method)
(ExcelFormerEncoder method)
(FeatureEncoder method)
(FTTransformerConvs method)
(LinearBucketEncoder method)
(LinearEmbeddingEncoder method)
(LinearEncoder method)
(LinearModelEncoder method)
(LinearPeriodicEncoder method)
(MultiCategoricalEmbeddingEncoder method)
(StackEncoder method)
(StypeEncoder method)
(TableConv method)
(TabTransformerConv method)
(TimestampEncoder method)
(TromptConv method)
(TromptDecoder method)
ResNet (class in torch_frame.nn.models)
RMSE (Metric attribute)
ROCAUC (Metric attribute)
S
save() (GBDT method)
(in module torch_frame.utils)
sequence_numerical (in module torch_frame.stype)
shuffle() (Dataset method)
split() (Dataset method)
StackEncoder (class in torch_frame.nn.encoder)
StatType (class in torch_frame.data)
STD (StatType attribute)
StypeEncoder (class in torch_frame.nn.encoder)
stypes (TensorFrame property)
StypeWiseFeatureEncoder (class in torch_frame.nn.encoder)
T
TableConv (class in torch_frame.nn.conv)
TabNet (class in torch_frame.nn.models)
TabTransformer (class in torch_frame.nn.models)
TabTransformerConv (class in torch_frame.nn.conv)
TabularBenchmark (class in torch_frame.datasets)
task_type (Dataset property)
TaskType (class in torch_frame.typing)
tensor_frame (Dataset property)
TensorFrame (class in torch_frame.data)
text_embedded (in module torch_frame.stype)
text_tokenized (in module torch_frame.stype)
TextEmbedderConfig (class in torch_frame.config)
TextTokenizerConfig (class in torch_frame.config)
timestamp (in module torch_frame.stype)
TimestampEncoder (class in torch_frame.nn.encoder)
Titanic (class in torch_frame.datasets)
to_dense() (MultiNestedTensor method)
torch_frame.stype
module
torch_frame.typing
module
transformed_stats (BaseTransform property)
Trompt (class in torch_frame.nn.models)
TromptConv (class in torch_frame.nn.conv)
TromptDecoder (class in torch_frame.nn.decoder)
tune() (GBDT method)
V
validate() (TensorFrame method)
X
XGBoost (class in torch_frame.gbdt)
Y
Yandex (class in torch_frame.datasets)
YEAR_RANGE (StatType attribute)
Z
ZEROS (NAStrategy attribute)
Read the Docs
v: latest
Versions
latest
stable
Downloads
On Read the Docs
Project Home
Builds