Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Fix code examples in PyTorch blog post - add missing imports and defi…
…ne variables

Co-authored-by: franciscojavierarceo <4163062+franciscojavierarceo@users.noreply.github.com>
  • Loading branch information
Copilot and franciscojavierarceo committed Jan 26, 2026
commit bf7f923577c4a16779b2ad52f6fad5a6c958c326
14 changes: 11 additions & 3 deletions infra/website/docs/blog/feast-joins-pytorch-ecosystem.md
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,8 @@ Feast integrates seamlessly with PyTorch workflows, whether you're training mode
from feast import FeatureStore
import torch
from torch.utils.data import Dataset, DataLoader
import pandas as pd
from datetime import datetime

# Initialize Feast
store = FeatureStore(repo_path=".")
Expand All @@ -58,7 +60,7 @@ training_df = store.get_historical_features(

# Create PyTorch Dataset
class FeatureDataset(Dataset):
def __init__(self, df):
def __init__(self, df, feature_cols):
self.features = torch.tensor(df[feature_cols].values, dtype=torch.float32)
self.labels = torch.tensor(df['label'].values, dtype=torch.float32)

Expand All @@ -69,10 +71,16 @@ class FeatureDataset(Dataset):
return self.features[idx], self.labels[idx]

# Train your PyTorch model
dataset = FeatureDataset(training_df)
feature_cols = ['age', 'activity_score', 'popularity']
dataset = FeatureDataset(training_df, feature_cols)
dataloader = DataLoader(dataset, batch_size=32, shuffle=True)

model = YourPyTorchModel()
# Define and train your model
model = torch.nn.Sequential(
torch.nn.Linear(len(feature_cols), 64),
torch.nn.ReLU(),
torch.nn.Linear(64, 1)
)
# Training loop...
```

Expand Down